gokaygokay commited on
Commit
15f3ff8
1 Parent(s): 985f78e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
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
- timestamp = int(time.time() * 1000) # Add this line
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>