Update app.py

app.py

@@ -1,3 +1,4 @@
+import
 import streamlit as st
 import numpy as np
 import pandas as pd
@@ -15,11 +16,11 @@ def get_model():
     #tokenizer = AutoTokenizer.from_pretrained("M4-ai/tau-1.8B")
     #model = AutoModelForCausalLM.from_pretrained("M4-ai/tau-1.8B")
 
-    tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM-360M")
-    model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM-360M")
+    #tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM-360M")
+    #model = AutoModelForCausalLM.from_pretrained("HuggingFaceTB/SmolLM-360M")
 
-    #tokenizer = AutoTokenizer.from_pretrained("facebook/opt-1.3b")
-    #model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln121Paraphrase")
+    tokenizer = AutoTokenizer.from_pretrained("facebook/opt-1.3b")
+    model = AutoModelForCausalLM.from_pretrained("BigSalmon/InformalToFormalLincoln121Paraphrase")
 
     #tokenizer = AutoTokenizer.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")
     #model = AutoModelForCausalLM.from_pretrained("TinyLlama/TinyLlama-1.1B-Chat-v1.0")