update model and tokenizer

app.py

@@ -2,9 +2,9 @@ import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
-model_name="fadliaulawi/polylm-1.7b-finetuned"
-model = AutoModelForCausalLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained("DAMO-NLP-MT/polylm-1.7b")
+model = PeftModel.from_pretrained(model, "fadliaulawi/polylm-1.7b-finetuned")
+tokenizer = AutoTokenizer.from_pretrained("DAMO-NLP-MT/polylm-1.7b")