Spaces:

prajjwal888
/

review-classifier

Sleeping

Prajjwal888 commited on Jun 14

Commit

fe69c7b

1 Parent(s): 605e349

update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -10,8 +10,8 @@ def predict_review(text):
     label = result['label']
     score = result['score']
-    if label == "AI":
-        rating = random.randint(1, 6)
     else:
         rating = random.randint(7, 10)

     label = result['label']
     score = result['score']
+    if label == "LABEL_0":
+        rating = random.randint(1, 4)
     else:
         rating = random.randint(7, 10)