Update app.py

app.py

@@ -6,21 +6,27 @@ import random
 hf_token = os.getenv("HF_TOKEN")
 
 # Load the pre-built text-to-image model interface
-interface = gr.load("models/ZB-Tech/Text-to-Image", token=hf_token)
+base_interface = gr.load("models/ZB-Tech/Text-to-Image", token=hf_token)
 
-# Function to generate a random seed and display it
-def generate_with_seed(prompt):
-    seed = random.randint(0, 1000000)  # Generate a random seed
-    # Note: The pre-built interface may not accept a seed parameter directly.
-    # This is just an example of how you might use it if you had control over the model.
-    return f"Generated with random seed: {seed}"
+# Function to generate an image with a random seed
+def generate_image_with_seed(prompt):
+    # Generate a random seed (for display or potential use)
+    seed = random.randint(0, 1000000)
+    
+    # Call the pre-built interface to generate the image
+    # Note: This assumes the interface takes a text prompt and returns an image
+    image = base_interface(prompt)
+    
+    # Return the image and the seed value
+    return image, f"Random seed used: {seed}"
 
-# Create a custom interface that includes the seed generation
+# Create a custom Gradio interface
 custom_interface = gr.Interface(
-    fn=generate_with_seed,
-    inputs="text",
-    outputs="text",  # Adjust this based on what the model actually returns (e.g., "image")
-    title="Text-to-Image with Random Seed"
+    fn=generate_image_with_seed,
+    inputs=gr.Textbox(label="Enter your prompt"),
+    outputs=[gr.Image(label="Generated Image"), gr.Textbox(label="Seed")],
+    title="Text-to-Image with Random Seed",
+    description="Generate an image from a text prompt with a random seed."
 )
 
 # Launch the interface