Update app.py

app.py

@@ -6,7 +6,8 @@ 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", use_auth_token=os.environ.get('sec_tok'))
+model = SetFitModel.from_pretrained("dvilasuero/setfit-mini-imdb")
 
 def classify(text):
     return model(text)