Update app.py

app.py

@@ -14,6 +14,8 @@ import random
 from torch.nn.parallel import DataParallel
 import os
 import matplotlib.pyplot as plt
+import gradio as gr
+import io
 
 class mydata(Dataset):
     def __init__(self, LR_path, GT_path, in_memory = True, transform = None):
@@ -593,4 +595,18 @@ def test_single_image(generator_path, input_image_path):
         plt.show()
 
 
-test_single_image('pre_trained_model_064.pt',uploaded_image_data)
+#test_single_image('pre_trained_model_064.pt',uploaded_image_data)
+# Define Gradio interface
+inputs = gr.inputs.Image(upload="single", label="Upload Image")
+output = gr.outputs.Image(label="Super-Resolved Image")
+
+# Load the generator model path
+generator_path = 'pre_trained_model_064.pt'
+
+
+
+# Deploy the interface
+gr.Interface(test_single_image, inputs, output, 
+              model_name="Super Resolution", 
+              title="Image Super-Resolution", 
+              description="Upload an image to see it super-resolved.").launch()