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 for ZeroGPU compatibility
+import spaces  
 
 # Load the model and processor
 model_path = "deepseek-ai/Janus-Pro-7B"
@@ -63,7 +63,8 @@ def generate(input_ids, width, height, cfg_weight=5, temperature=1.0, parallel_s
     return patches
 
 def unpack(patches, width, height, parallel_size=5):
-    patches = patches.to(torch.float32).cpu().numpy().transpose(0, 2, 3, 1)
+    # Detach the tensor before converting to numpy
+    patches = patches.detach().to(torch.float32).cpu().numpy().transpose(0, 2, 3, 1)
     patches = np.clip((patches + 1) / 2 * 255, 0, 255)
 
     images = [Image.fromarray(patches[i].astype(np.uint8)) for i in range(parallel_size)]
@@ -128,5 +129,20 @@ def create_interface():
 
     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%2Fradiogold">
+        <img src="https://api.visitorbadge.io/api/visitors?path=https%3A%2F%2Fhuggingface.co%2Fspaces%2FBils%2Fradiogold&countColor=%23263759" />
+    </a>
+    """)
+
 demo = create_interface()
 demo.launch(share=True)