p4vv37 commited on
Commit
02dbbde
·
1 Parent(s): 5c2d612

updated requirements

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ def generate_view(source_img, elevation, azimuth, camera_distance, num_inference
13
  pipeline.to('cuda:0')
14
 
15
  # Prepare input data:
16
- image = source_img.resize((256, 256))
17
 
18
 
19
  # Generate and save images:
@@ -26,7 +26,7 @@ def generate_view(source_img, elevation, azimuth, camera_distance, num_inference
26
  return images[0]
27
 
28
 
29
- iface = gr.Interface(fn=generate_view, inputs=[gr.Image(type="pil", mode="RGB", value="images/bottle.png"),
30
  gr.Number(label="elevation", value=0.),
31
  gr.Number(label="azimuth", value=45.),
32
  gr.Number(label="camera_distance", value=1.2),
 
13
  pipeline.to('cuda:0')
14
 
15
  # Prepare input data:
16
+ image = source_img.resize((256, 256)).convert("RGB")
17
 
18
 
19
  # Generate and save images:
 
26
  return images[0]
27
 
28
 
29
+ iface = gr.Interface(fn=generate_view, inputs=[gr.Image(type="pil", value="images/bottle.png"),
30
  gr.Number(label="elevation", value=0.),
31
  gr.Number(label="azimuth", value=45.),
32
  gr.Number(label="camera_distance", value=1.2),