Update app.py

app.py

@@ -14,10 +14,11 @@ models = {
 selected_model = st.radio("Select Model", list(models.keys()))
 model_name = models[selected_model]
 tokenizer = AutoTokenizer.from_pretrained('t5-small')
+model = torch.load(model_name+'/model.pt')
 
-if model_name == "T5 Small":
+#if model_name == "T5 Small":
     #model = T5().from_pretrained(model_name)
-    model = torch.load(model_name+'/model.pt')
+    #model = torch.load(model_name+'/model.pt')
 
     
 #else: