Update app.py

app.py

@@ -26,8 +26,10 @@ def load_model():
     nltk.download('stopwords')
     nltk.download('wordnet')
     nltk.download('omw-1.4')
-    ##summary_mod_name = os.environ["summary_mod_name"]
-    question_mod_name = os.environ["question_mod_name"]
+    ## summary_mod_name = os.environ["summary_mod_name"]
+    ## question_mod_name = os.environ["question_mod_name"]
+    summary_mod_name = "t5-small"
+    summary_mod_name= "t5-small"
     summary_model = T5ForConditionalGeneration.from_pretrained(summary_mod_name)
     summary_tokenizer = T5Tokenizer.from_pretrained(summary_mod_name)
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -51,7 +53,7 @@ def load_file():
         return raw_text
 
 # Loading Model
-summary_model, summary_tokenizer, question_tokenizer, question_model =load_model()
+summary_model, summary_tokenizer, question_tokenizer, question_model = load_model()
 
 # App title and description
 st.title("P's Prototype")