suyoyog commited on
Commit
0cbd007
·
verified ·
1 Parent(s): db9637c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -597,8 +597,8 @@ def test_single_image(generator_path, input_image_path):
597
 
598
  #test_single_image('pre_trained_model_064.pt',uploaded_image_data)
599
  # Define Gradio interface
600
- inputs = gr.inputs.Image(upload="single", label="Upload Image")
601
- output = gr.outputs.Image(label="Super-Resolved Image")
602
 
603
  # Load the generator model path
604
  generator_path = 'pre_trained_model_064.pt'
 
597
 
598
  #test_single_image('pre_trained_model_064.pt',uploaded_image_data)
599
  # Define Gradio interface
600
+ inputs = gr.inputs.Image(type="pil", label="Upload Image")
601
+ output = gr.outputs.Image(type="file", label="Super-Resolved Image")
602
 
603
  # Load the generator model path
604
  generator_path = 'pre_trained_model_064.pt'