Update app.py

app.py

@@ -18,7 +18,10 @@ classifier = pipeline("text-classification", tokenizer=tokenizer, model=model, m
 
 def prozesatu(testua):
     prediction = prozesatu.classifier(testua)[0]
-    return {"Gai":prediction["score"], "Ez gai": 1-prediction["score"]}
+    if prediction["label"]=="GAI":
+        return {"Gai":prediction["score"], "Ez gai": 1-prediction["score"]}
+    else:
+        return {"Gai":1-prediction["score"], "Ez gai": prediction["score"]}
     #return 'C1:{}. Probabilitatea:{:.2f}'.format(prediction["label"], round(prediction["score"], 2))
 
 prozesatu.classifier = classifier