Update app.py

app.py

@@ -7,7 +7,7 @@ import json
 @st.cache()
 def get_model():
     model = AutoModelForSequenceClassification.from_pretrained("siebert/sentiment-roberta-large-english", num_labels=2)
-    model.load_state_dict(torch.load('model'))
+    model.load_state_dict(torch.load('cached_model.pth'))
     return model
 
 @st.cache()
@@ -15,9 +15,11 @@ def get_tokenizer():
     tokenizer = AutoTokenizer.from_pretrained("siebert/sentiment-roberta-large-english")
     return tokenizer
 
-def make_prediction():
+def make_prediction(to_analyze):
     model = get_model()
     tokenizer = tokenizer()
+    to_return = model(**tokenizer(to_anayze))
+    return to_return
     
 
 
@@ -41,7 +43,8 @@ with st.form(key='input_form'):
     button = st.form_submit_button(label='Classify')
 if button:
     if to_analyze:
-        make_prediction(to_analyze)
+        pred = make_prediction(to_analyze)
+        st.markdown(pred)
     else:
         st.markdown("Empty request. Please resubmit")