Update app.py

app.py

@@ -31,6 +31,8 @@ def load_img(filename):
 
 def process_img(image, filename):
     # Select the correct weight file based on the image filename
+
+    filename = image.name.split("/")[-1]
     if filename in image_to_weights:
         model_path = image_to_weights[filename]
         checkpoints = torch.load(model_path, map_location=device)
@@ -90,17 +92,10 @@ demo = gr.Interface(
     css=css
 )
 
-# Updating the filename in the input after selection
-def update_filename(image):
-    # Retrieve the filename from the input image
-    if image:
-        filename = image.filename  # Gradio automatically gives the file name
-        return filename
-    return ""
 
 with gr.Blocks() as demo:
     with gr.Row():
-        input_image = gr.Image(type='pil', label='input')
+        input_image = gr.Image(type='pil', label='input', interactive = True)
         filename_output = gr.Textbox(label="Image Filename", interactive=False)
 
     # Define the output