Update app.py

app.py

@@ -26,6 +26,26 @@ def train_model():
     
     return f"Modelo treinado e salvo em: {model_filename}"
 
+# Carregar o modelo treinado (supondo que você tenha o arquivo 'model.pkl')
+model = load("/mnt/data/model.pkl")  # ou o caminho do seu modelo
+
+# Função para fazer previsões
+def predict(input_text):
+    # Prever usando o modelo carregado
+    prediction = model.predict([input_text])  # ou adapte conforme o seu modelo
+    return prediction[0]
+
+# Criar a interface Gradio
+iface = gr.Interface(
+    fn=predict,                  # Função para chamar ao gerar
+    inputs=gr.Textbox(label="Input Text"),  # Caixa de entrada de texto
+    outputs=gr.Textbox(label="Output")     # Caixa de saída com a previsão
+)
+
+# Iniciar a aplicação
+iface.launch()
+
+
 # Inicialize a interface Gradio
 iface = gr.Interface(fn=train_model, inputs=[], outputs=["text"])