Update app.py

app.py

@@ -6,7 +6,6 @@ from smolagents import DuckDuckGoSearchTool, Tool
 from diffusers import DiffusionPipeline
 import torch
 from smolagents import OpenAIServerModel
-
 import os
 from huggingface_hub import login
 
@@ -60,7 +59,7 @@ def add_label_to_image(image, label):
 
 def plot_and_save_agent_image(agent_image, label, save_path=None):
     #pil_image = agent_image.to_raw()
-    pil_image=agent_image
+    pil_image = agent_image
     labeled_image = add_label_to_image(pil_image, label)
     #labeled_image.show()
     if save_path:
@@ -78,6 +77,11 @@ def generate_prompts_for_object(object_name):
     }
 
 
+
+# =========================================================
+# Tool and Agent Initialization
+# =========================================================
+
 image_generation_tool = Tool.from_space(
     #"KingNish/Realtime-FLUX",
     "black-forest-labs/FLUX.1-schnell",
@@ -86,15 +90,10 @@ image_generation_tool = Tool.from_space(
     description="Generate an image from a prompt"
 )
 
-
-# =========================================================
-# Tool and Agent Initialization
-# =========================================================
-
 search_tool = DuckDuckGoSearchTool()
 #llm_engine = InferenceClientModel("Qwen/Qwen2.5-72B-Instruct")
 
-#llm_engine = InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct")
+llm_engine2 = InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct", provider="together")
 
 # Inicialização do modelo OpenAI com smolagents
 llm_engine = OpenAIServerModel(
@@ -173,6 +172,7 @@ def generate_object_history(object_name):
 # Gradio Interface
 # =========================================================
 
+
 def create_gradio_interface():
     with gr.Blocks() as demo:
         gr.Markdown("# TimeMetamorphy: An Object Evolution Generator")
@@ -182,9 +182,9 @@ def create_gradio_interface():
         """)
 
         default_images = [
-            "car_past.png",
-            "car_present.png",
-            "car_future.png"
+            "car_past2.png",
+            "car_present2.png",
+            "car_future2.png"
         ]
         default_gif_path = "car_evolution.gif"
 
@@ -192,10 +192,10 @@ def create_gradio_interface():
             with gr.Column():
                 object_name_input = gr.Textbox(label="Enter an object name", placeholder="e.g. bicycle, car, phone")
                 generate_button = gr.Button("Generate Evolution")
-                #image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, value=default_images, type="filepath")
-                #gif_output = gr.Image(label="Generated GIF", value=default_gif_path, type="filepath")
-                image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1,  type="filepath")
-                gif_output = gr.Image(label="Generated GIF", type="filepath")
+                image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1, value=default_images, type="filepath")
+                gif_output = gr.Image(label="Generated GIF", value=default_gif_path, type="filepath")
+                #image_gallery = gr.Gallery(label="Generated Images", columns=3, rows=1,  type="filepath")
+                #gif_output = gr.Image(label="Generated GIF", type="filepath")
 
         generate_button.click(fn=generate_object_history, inputs=[object_name_input], outputs=[image_gallery, gif_output])