update

app.py

@@ -1,4 +1,4 @@
-import spaces
+#import spaces
 import sys
 import time
 import os
@@ -33,15 +33,15 @@ unet3d_condition_model_forward_copy = UNet3DConditionModel.forward
 UNet3DConditionModel.forward = unet3d_condition_model_forward
 
 
-model_id = "cerspense/zeroscope_v2_576w"
-model_path = model_id
-pipe = DiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
-pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
-pipe.to("cuda")
+# model_id = "cerspense/zeroscope_v2_576w"
+# model_path = model_id
+# pipe = DiffusionPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
+# pipe.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
+# pipe.to("cuda")
 
 MAX_KEYS = 10
 
-@spaces.GPU(duration=120)
+#@spaces.GPU(duration=120)
 def core(bundle):
     generator = torch.Generator().manual_seed(int(bundle["seed"]))
     result = pipe(
@@ -225,12 +225,18 @@ with gr.Blocks(
     <a href="https://hits.seeyoufarm.com"><img src="https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fhohonu-vicml%2FTrailblazer&count_bg=%23FF3232&title_bg=%23555555&icon=&icon_color=%23E7E7E7&title=hits&edge_flat=false"></a>
     """
     with gr.Row():
-        gr.HTML("""<a href="https://arxiv.org/abs/2401.00896"><img src="https://img.shields.io/badge/cs.CV-Paper-b31b1b?logo=arxiv&logoColor=red"></a>""")
-        gr.HTML("""<a href="https://hohonu-vicml.github.io/Trailblazer.Page/"><img src="https://img.shields.io/badge/TrailBlazer-Website-green?logo=googlechrome&logoColor=green"></a>""")
-        gr.HTML("""<a href="https://github.com/hohonu-vicml/Trailblazer"><img src="https://img.shields.io/badge/Github-Code-green?logo=github&logoColor=black"></a>""")
-        gr.HTML("""<a href="https://www.youtube.com/watch?v=kEN-32wN-xQ"><img src="https://img.shields.io/badge/YouTube-Project-c4302b?logo=youtube&logoColor=red"></a>""")
-        gr.HTML("""<a href="https://www.youtube.com/watch?v=P-PSkS7sNco"><img src="https://img.shields.io/badge/YouTube-Result-c4302b?logo=youtube&logoColor=red"></a>""")
-        gr.HTML("""<a href="https://hits.seeyoufarm.com"><img src="https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fhohonu-vicml%2FTrailblazer&count_bg=%23FF3232&title_bg=%23555555&icon=&icon_color=%23E7E7E7&title=hits&edge_flat=false"></a>""")
+        with gr.Column(scale=1):
+            pass
+        with gr.Column(scale=4):
+            with gr.Row():
+                gr.HTML("""<a href="https://arxiv.org/abs/2401.00896"><img src="https://img.shields.io/badge/cs.CV-Paper-b31b1b?logo=arxiv&logoColor=red"></a>""")
+                gr.HTML("""<a href="https://hohonu-vicml.github.io/Trailblazer.Page/"><img src="https://img.shields.io/badge/TrailBlazer-Website-green?logo=googlechrome&logoColor=green"></a>""")
+                gr.HTML("""<a href="https://github.com/hohonu-vicml/Trailblazer"><img src="https://img.shields.io/badge/Github-Code-green?logo=github&logoColor=black"></a>""")
+                gr.HTML("""<a href="https://www.youtube.com/watch?v=kEN-32wN-xQ"><img src="https://img.shields.io/badge/YouTube-Project-c4302b?logo=youtube&logoColor=red"></a>""")
+                gr.HTML("""<a href="https://www.youtube.com/watch?v=P-PSkS7sNco"><img src="https://img.shields.io/badge/YouTube-Result-c4302b?logo=youtube&logoColor=red"></a>""")
+                gr.HTML("""<a href="https://hits.seeyoufarm.com"><img src="https://hits.seeyoufarm.com/api/count/incr/badge.svg?url=https%3A%2F%2Fhuggingface.co%2Fspaces%2Fhohonu-vicml%2FTrailblazer&count_bg=%23FF3232&title_bg=%23555555&icon=&icon_color=%23E7E7E7&title=hits&edge_flat=false"></a>""")
+        with gr.Column(scale=1):
+            pass
 
     description = """
     <p>