Update app.py

app.py

@@ -13,14 +13,23 @@ def classify_image_with_text(text, image):
     return {label: score for label, score in zip(labels, scores)}
 
 # Créer l'interface Gradio
-iface = gr.Interface(
-    fn=classify_image_with_text,
-    inputs=[
-        gr.inputs.Textbox(lines=5, label="Texte"),
-        gr.inputs.Image(label="Image")
-    ],
-    outputs=gr.outputs.Label(num_top_classes=3)  # Ajustez le nombre de classes principales si nécessaire
-)
+with gr.Blocks(title="SD Models") as my_interface:
+    with gr.Column(scale=12):
+        with gr.Row():
+            with gr.Row(scale=6):
+                primary_prompt = gr.Textbox(label="Prompt", value="")
+                input_image = gr.Image(label="Image")
+
+            with gr.Row(scale=6):
+                with gr.Row():
+                    api = gr.Button("Api", variant="primary")
+        with gr.Row():
+            api_image_output = gr.Textbox(label='Api OutPut')
+         
+
+        api.click(classify_image_with_text, inputs=[primary_prompt,input_image], outputs=[api_image_output], api_name='generate')
+
+
 
 # Lancer l'interface
 iface.launch()