Spaces:

gokaygokay
/

360PanoImage

Running on Zero

gokaygokay commited on 9 days ago

Commit

e746ea4

•

1 Parent(s): eb5cd64

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ model_path = snapshot_download("archerfmy0831/sd-t2i-360panoimage")
 # Initialize pipelines
 txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float16)
-@spaces.GPU(duration=200)
 def text_to_pano(prompt, upscale):
     input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
     output = txt2panoimg(input_data)

 # Initialize pipelines
 txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float16)
+@spaces.GPU
 def text_to_pano(prompt, upscale):
     input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
     output = txt2panoimg(input_data)