Update app.py

app.py

@@ -8,17 +8,17 @@ import gradio as gr
 #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)[0]['label']
+    return classifier(txt)[0]['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 .',