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)['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 .',