Update app.py

app.py

@@ -1,24 +1,24 @@
 import pandas as pd
 from fastai.text.all import *
 from datasets import load_dataset
-#from transformers import pipeline
+from transformers import pipeline
 import gradio as gr
 
 # Modelo de clasificación usando LSTM
 #learner = load_learner('modelLSTM.pkl')
 
 # Modelo de clasificación de texto usando modelos de lenguaje
-learner = load_learner('modelML.pkl')
+#learner = load_learner('modelML.pkl')
 
 # Modelo de clasificación basados en mecanismos de atención
-#classifier = pipeline('text-classification', model='edgilr/clasificador-rotten-tomatoes')
+classifier = pipeline('text-classification', model='edgilr/clasificador-rotten-tomatoes')
 
 def predict(txt):
     # Modelo de clasificación usando LSTM o modelo de clasificación de texto usando modelos de lenguaje
-    pred,pred_idx,probs = learner.predict(txt)
-    return pred
+    #pred,pred_idx,probs = learner.predict(txt)
+    #return pred
     # Modelo de clasificación basados en mecanismos de atención
-    #return classifier(txt)['label']
+    return classifier(txt)['label']
     
 gr.Interface(fn=predict, inputs="text", outputs="text",
              examples=['the story gives ample opportunity for large-scale action and suspense , which director shekhar kapur supplies with tremendous skill .',