Update app.py

app.py

@@ -9,8 +9,8 @@ from diffusers import DiffusionPipeline
 from huggingface_hub import login
 import numpy as np
 
-def guessanImage(image):
-    model = "microsoft/resnet-50"
+def guessanImage(image, model):
+    #model = "microsoft/resnet-50"
     # st.write("using model:"+model)
     
     imgclassifier  = pipeline("image-classification", model=model)
@@ -18,9 +18,11 @@ def guessanImage(image):
         description = imgclassifier(image)
     return description
 
+radio = gr.Radio(["microsoft/resnet-50", "google/vit-base-patch16-224", "apple/mobilevit-small"], label="Select a Classifier", info="Image Classifier?"),
+
 demo = gr.Interface(
     fn=guessanImage,
-    inputs=[gr.Image(type="pil")],
+    inputs=[gr.Image(type="pil"), radio],
     outputs=["text"],
 )