Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -5,7 +5,6 @@ import torch
|
|
5 |
from huggingface_hub import snapshot_download
|
6 |
from txt2panoimg import Text2360PanoramaImagePipeline
|
7 |
from PIL import Image
|
8 |
-
import time
|
9 |
|
10 |
# Download the model
|
11 |
model_path = snapshot_download("archerfmy0831/sd-t2i-360panoimage")
|
@@ -17,8 +16,7 @@ txt2panoimg = Text2360PanoramaImagePipeline(model_path, torch_dtype=torch.float1
|
|
17 |
def text_to_pano(prompt, upscale):
|
18 |
input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
|
19 |
output = txt2panoimg(input_data)
|
20 |
-
|
21 |
-
return f"{output}?t={timestamp}", output # Modify this line
|
22 |
|
23 |
title = """<h1 align="center">SD-T2I-360PanoImage</h1>
|
24 |
<p align="center">360° Panorama Image Generation</p>
|
|
|
5 |
from huggingface_hub import snapshot_download
|
6 |
from txt2panoimg import Text2360PanoramaImagePipeline
|
7 |
from PIL import Image
|
|
|
8 |
|
9 |
# Download the model
|
10 |
model_path = snapshot_download("archerfmy0831/sd-t2i-360panoimage")
|
|
|
16 |
def text_to_pano(prompt, upscale):
|
17 |
input_data = {'prompt': prompt, 'upscale': upscale, 'refinement': False}
|
18 |
output = txt2panoimg(input_data)
|
19 |
+
return output, output
|
|
|
20 |
|
21 |
title = """<h1 align="center">SD-T2I-360PanoImage</h1>
|
22 |
<p align="center">360° Panorama Image Generation</p>
|