Update app.py

app.py

@@ -5,20 +5,20 @@ from transformers import pipeline
 import gradio as gr
 
 # Modelo de clasificación usando LSTM
-#learner = load_learner('modelLSTM.pkl')
+learner = load_learner('modelLSTM.pkl')
 
 # Modelo de clasificación de texto usando modelos de lenguaje
 #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 .',