Update app.py

app.py

@@ -47,8 +47,10 @@ preprocess = transforms.Compose([
 ])
 
 def classify_image(image, model_name):
-    model = load_model(model_name)
+    if image is None:
+        return "Please upload an image."
     
+    model = load_model(model_name)
     image = Image.fromarray(image)
     input_image = preprocess(image).unsqueeze(0).to(device)
     
@@ -61,7 +63,7 @@ def classify_image(image, model_name):
 
 # Gradio interface
 image_input = gr.Image(image_mode="RGB")
-model_choice = gr.Radio(choices=list(model_options.keys()), label="Choose Model", value="ResNet-18")
+model_choice = gr.Dropdown(choices=list(model_options.keys()), label="Choose Model", value="ResNet-18")
 output_text = gr.Textbox()
 
 gr.Interface(fn=classify_image, inputs=[image_input, model_choice], outputs=[output_text],