Update app.py

app.py

@@ -89,6 +89,7 @@ with st.spinner('Loading classification model...'):
     from transformers import pipeline
 
     checkpoint = "amir7d0/distilbert-base-uncased-finetuned-amazon-reviews"
+    checkpoint = "/home/v4vendetta/Documents/bert-fa-base-uncased-sentiment-digikala"
     classifier = pipeline("text-classification", model=checkpoint)
 
 
@@ -145,16 +146,17 @@ with tab2:
                 p_time = round(end_time-start_time, 2)
                 st.success(f'Prediction finished in {p_time}s!')
 
+                for text, pred in zip(texts, preds):
+                    pred['text'] = text
+
             c1, c2 = st.columns([3, 1])
             with c1:
                 st.subheader("🎈 Check & download results")
             with c2:
-                CSVButton2 = download_button(results, "Data.csv", "📥 Download (.csv)")
+                CSVButton2 = download_button(preds, "sentiment-analysis-preds.csv", "📥 Download (.csv)")
 
             st.header("")
 
-            for text, pred in zip(texts, preds):
-                pred['text'] = text
 
             df = pd.DataFrame(preds, columns=['text', 'label', 'score'])