Update app.py

app.py

@@ -1,8 +1,27 @@
 import gradio as gr
 import os
+import random
 
 # Load the API token from the environment variable
 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)
 
-interface.launch()
+# 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}"
+
+# Create a custom interface that includes the seed generation
+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"
+)
+
+# Launch the interface
+custom_interface.launch()