Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 
-# Definir informações adicionais
+gr.Interface.load("models/DHEIVER/Advances-in-Breast-Ultrasound").launch()
+
+# Defina informações sobre o modelo
 model_info = """
 **Modelo de Ultrassom Avançado para Mama**
 
@@ -37,10 +39,17 @@ Durante o treinamento, os seguintes hiperparâmetros foram utilizados:
 - Número de Épocas (num_epochs): 14
 """
 
-# Criar a interface com informações do modelo
+# Defina a interface Gradio
+input_text = gr.Textbox()
+output_text = gr.Textbox()
+
 model_interface = gr.Interface(
-    fn="models/DHEIVER/Advances-in-Breast-Ultrasound", 
-    description=model_info
+    fn=predict, 
+    inputs=input_text,
+    outputs=output_text,
+    title="Modelo de Ultrassom Avançado para Mama",
+    description=model_info,
+    live=True
 )
 
 # Lançar a interface