Update app.py

app.py

@@ -2,9 +2,20 @@ import gradio as gr
 import torch
 from transformers import AutoFeatureExtractor, AutoModelForImageClassification
 
-def aiornot(img):    
+models=[
+    "Nahrawy/AIorNot",
+    "RishiDarkDevil/ai-image-det-resnet152",
+    "arnolfokam/ai-generated-image-detector",
+]
+
+
+
+def aiornot(img,mod_choose):    
     labels = ["Real", "AI"]
-    feature_extractor = AutoFeatureExtractor.from_pretrained("Nahrawy/AIorNot")
+    
+    #feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
+    mod=models[int(mod_choose)]
+    feature_extractor = AutoFeatureExtractor.from_pretrained(models[])
     model = AutoModelForImageClassification.from_pretrained("Nahrawy/AIorNot")
     
     input = feature_extractor(image, return_tensors="pt")
@@ -17,8 +28,10 @@ def aiornot(img):
 
 with gr.Blocks() as app:
     with gr.Row():
-        inp = gr.Image()
-        btn = gr.Button()
+        with gr.Column():
+            inp = gr.Image()
+            mod_choose=gr.Number(value=0)
+            btn = gr.Button()
     outp = gr.Textbox()
-    btn.click(aiornot,inp,outp)
+    btn.click(aiornot,[inp,mod_choose],outp)
 app.launch()