Update app.py

app.py

@@ -16,18 +16,18 @@ import gradio as gr
 #model = learn.model
 #model = model.cpu()
 
-#repo_id = "maviced/practica7-2"
-#learn = from_pretrained_fastai(repo_id)
-#model = learn.model
-#model = model.cpu()
+repo_id = "maviced/practica7-2"
+learn = from_pretrained_fastai(repo_id)
+model = learn.model
+model = model.cpu()
 
-classifier = pipeline('text-classification', model='maviced/clasificador-rotten-tomatoes')
+#classifier = pipeline('text-classification', model='maviced/clasificador-rotten-tomatoes')
 
 def predict(text):
-    #pred,pred_idx,probs = learner.predict(text)
-    #return pred
+    pred,pred_idx,probs = model.predict(text)
+    return pred
     
-    return classifier(text)[0]['label']
+    #return classifier(text)[0]['label']
     
 gr.Interface(fn=predict, inputs="text", outputs="text",examples=['A visually stunning and emotionally resonant film that leaves a lasting impact.',
                        'The film, while visually stunning, falls short with a predictable plot and underdeveloped characters, leaving the audience craving more substance.']).launch(share=True)