updated code

app.py

@@ -310,7 +310,7 @@ def app():
     
     # Left column: After clicking "Run Model" button, display the final output style transfer image
     with right_col:
-        if uploaded_image is not None and selected_image in style_dict:
+        if uploaded_image is not None and selected_image in models_dict:
             # Get the selected style image
             # style_image_path = style_images[selected_model]