Spaces:

gokaygokay
/

360PanoImage

Running on Zero

gokaygokay commited on Jul 9, 2024

Commit

b50c00b

verified ·

1 Parent(s): 5bac20d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -14,7 +14,7 @@ txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float1
 @spaces.GPU(duration=200)
 def text_to_pano(prompt, upscale):
-    input_data = {'prompt': prompt, 'upscale': upscale}
     output = txt2panoimg(input_data)
     return output, output

 @spaces.GPU(duration=200)
 def text_to_pano(prompt, upscale):
+    input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
     output = txt2panoimg(input_data)
     return output, output