Update app.py

app.py

@@ -18,8 +18,11 @@ def get_model():
     #tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM-360M")
     #model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM-360M")
     
-    tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-zephyr-1_6b")
-    model = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-2-zephyr-1_6b")
+    #tokenizer = AutoTokenizer.from_pretrained("stabilityai/stablelm-2-zephyr-1_6b")
+    #model = AutoModelForCausalLM.from_pretrained("stabilityai/stablelm-2-zephyr-1_6b")
+
+    tokenizer = AutoTokenizer.from_pretrained("BigSalmon/InformalToFormalLincoln122Paraphrase")
+    model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln122Paraphrase")
 
     #tokenizer = AutoTokenizer.from_pretrained("facebook/opt-1.3b")
     #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln121Paraphrase")