Update app.py

app.py

@@ -7,23 +7,11 @@ repo_id1 = "nlmaldonadog/AWD_LSTM_P7"
 learner1 = from_pretrained_fastai(repo_id1)
 labels1 = learner1.dls.vocab
 
-# Cargar el segundo modelo
-repo_id2 = "nlmaldonadog/AWD_LSTM_P7_LM"
-learner2 = from_pretrained_fastai(repo_id2)
-labels2 = learner2.dls.vocab
-
-def predict(text):
-    # Predicciones del primer modelo
-    pred1, pred_idx1, probs1 = learner1.predict(text)
-    result1 = {labels1[i]: float(probs1[i]) for i in range(len(labels1))}
-    
-    # Predicciones del segundo modelo
-    pred2, pred_idx2, probs2 = learner2.predict(text)
-    result2 = {labels2[i]: float(probs2[i]) for i in range(len(labels2))}
-    
-    return {"Modelo 1": result1, "Modelo 2": result2}
+def predict1(text):
+    pred, pred_idx, probs = learner1.predict(text)
+    return {labels1[i]: float(probs[i]) for i in range(len(labels1))}
 
 texto = gr.inputs.Textbox(lines=2, placeholder='Escribe aquí...')
 
 # Creamos la interfaz y la lanzamos. 
-gr.Interface(fn=predict, inputs=texto, outputs="json").launch(share=False)
+gr.Interface(fn=predict1, inputs=texto, outputs=gr.outputs.Label()).launch(share=False)