Update app.py
Browse files
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(
|
| 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'
|