Update app.py

app.py

@@ -6,11 +6,14 @@ from setfit import SetFitModel
 def greet(name):
     return "Hello " + name + "!!"
 
-# model = SetFitModel.from_pretrained("AISE-TUDelft/java-ownership-classifier", use_auth_token=os.environ.get('sec_tok'))
+# model = SetFitModel.from_pretrained("AISE-TUDelft/java-ownership-classifier")
 model = SetFitModel.from_pretrained("dvilasuero/setfit-mini-imdb")
 
 def classify(text):
-    return model([text])[0]
+    if model([text])[0]:
+        return 'True'
+    else:
+        return 'False'
     
 iface = gr.Interface(fn=classify, inputs="text", outputs="text")
 iface.launch()