Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoConfig, AutoModelForCausalLM
 from janus.models import MultiModalityCausalLM, VLChatProcessor
 from PIL import Image
 import numpy as np
-import spaces  
+import spaces  # Import spaces for ZeroGPU compatibility
 
 # Load the model and processor
 model_path = "deepseek-ai/Janus-Pro-7B"
@@ -114,9 +114,12 @@ def create_interface():
         """)
         
         prompt_input = gr.Textbox(label="Prompt (describe the image)")
-        seed_input = gr.Number(label="Seed (Optional)", value=12345, precision=0)
-        guidance_slider = gr.Slider(label="CFG Guidance Weight", minimum=1, maximum=10, value=5, step=0.5)
-        temperature_slider = gr.Slider(label="Temperature", minimum=0, maximum=1, value=1.0, step=0.05)
+
+        # Option to toggle additional parameters
+        with gr.Accordion("Advanced Parameters", open=False):
+            seed_input = gr.Number(label="Seed (Optional)", value=12345, precision=0)
+            guidance_slider = gr.Slider(label="CFG Guidance Weight", minimum=1, maximum=10, value=5, step=0.5)
+            temperature_slider = gr.Slider(label="Temperature", minimum=0, maximum=1, value=1.0, step=0.05)
 
         generate_button = gr.Button("Generate Images")
         output_gallery = gr.Gallery(label="Generated Images", columns=2, height=300)
@@ -127,22 +130,20 @@ def create_interface():
             outputs=output_gallery
         )
 
-    return demo
+        # Footer
+        gr.Markdown("""
+        <hr>
+        <p style="text-align: center; font-size: 0.9em;">
+            Created with ❤️ by <a href="https://bilsimaging.com" target="_blank">bilsimaging.com</a>
+        </p>
+        """)
+        
+        # Visitor Badge
+        gr.HTML("""
+        <a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FBils%2FDeepseekJanusPro%2F"><img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FBils%2FDeepseekJanusPro%2F&countColor=%23263759" /></a>
+        """)
 
-    # Footer
-    gr.Markdown("""
-    <hr>
-    <p style="text-align: center; font-size: 0.9em;">
-        Created with ❤️ by <a href="https://bilsimaging.com" target="_blank">bilsimaging.com</a>
-    </p>
-    """)
-    
-    # Visitor Badge
-    gr.HTML("""
-    <a href="https://visitorbadge.io/status?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FBils%2Fradiogold">
-        <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FBils%2Fradiogold&countColor=%23263759" />
-    </a>
-    """)
+    return demo
 
 demo = create_interface()
 demo.launch(share=True)