Manjushri commited on
Commit
074954d
·
1 Parent(s): 14f8699

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -18,11 +18,11 @@ input_image = gr.inputs.Image(type="filepath")
18
  output_image = gr.outputs.Image(type="filepath")
19
 
20
  # Define the function that will be called when the interface is used
21
- def upscale_image(model, image):
22
  # Convert the image to a PyTorch tensor
23
  generator = torch.manual_seed(999999)
24
- low_res_img = Image.open(low_res_img).convert("RGB")
25
- low_res_latents = low_res_img
26
 
27
  # Upscale the image using the selected model
28
  if model == "SD 2.0 2x Latent Upscaler":
 
18
  output_image = gr.outputs.Image(type="filepath")
19
 
20
  # Define the function that will be called when the interface is used
21
+ def upscale_image(model, input_image):
22
  # Convert the image to a PyTorch tensor
23
  generator = torch.manual_seed(999999)
24
+ image = Image.open(input_image).convert("RGB")
25
+
26
 
27
  # Upscale the image using the selected model
28
  if model == "SD 2.0 2x Latent Upscaler":