Upload app.py

app.py

@@ -143,7 +143,7 @@ class NLP:
             percent, score = self.__exec[model][1](review)
             res = pd.DataFrame({'Rated 1/5': percent[0][0], 'Rated 2/5': percent[0][1], 'Rated 4/5': percent[0][2], 'Rated 5/5': percent[0][3]}, index=["Prediction"])
 
-        if (percent[0][0] == 0 and percent[1][0] == 0):
+        if (percent[0][0] == 0 and percent[0][1] == 0):
             return (res, f"Model: {model}\nDataset: {Dataset}\nAccuracy: {str(float(score)*100)}\nDue to the size of the model, it has not been implemented on huggingface.")
         else:
             return (res, f"Model: {model}\nDataset: {Dataset}\nAccuracy: {str(float(score)*100)}")