Update app.py

app.py

@@ -10,8 +10,7 @@ tokenizer = AutoTokenizer.from_pretrained('bigscience/bloom')
 
 BASE_MODEL = "bigscience/bloom-3b"
 
-#LORA_WEIGHTS = f"/content/drive/MyDrive/Colab Notebooks/LegalChatbot-{model_name}"
-LORA_WEIGHTS = f"jslin09/LegalChatbot-bloom-3b"
+LORA_WEIGHTS = "jslin09/LegalChatbot-bloom-3b"
 
 config = PeftConfig.from_pretrained(LORA_WEIGHTS)
 
@@ -112,7 +111,7 @@ def evaluate(
     max_new_tokens=128,
     **kwargs,
 ):
-    prompt = generate_prompt_tw(instruction, input) # 中文版的話，函數名稱要改用 generate_prompt_tw
+    prompt = generate_prompt(instruction, input) # 中文版的話，函數名稱要改用 generate_prompt_tw
     inputs = tokenizer(prompt, return_tensors="pt")
     input_ids = inputs["input_ids"].to(device)
     generation_config = GenerationConfig(