Update app.py

app.py

@@ -15,7 +15,7 @@ def aiornot(img,mod_choose):
     
     #feature_extractor = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
     mod=models[int(mod_choose)]
-    feature_extractor = AutoFeatureExtractor.from_pretrained(models[])
+    feature_extractor = AutoFeatureExtractor.from_pretrained(f'{mod}')
     model = AutoModelForImageClassification.from_pretrained("Nahrawy/AIorNot")
     
     input = feature_extractor(image, return_tensors="pt")