Update app.py

app.py

@@ -122,7 +122,8 @@ def aiornot1(image):
 def aiornot2(image):    
     labels = ["Real", "AI"]
     mod=models[2]
-    feature_extractor2 = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
+    feature_extractor2 = AutoFeatureExtractor.from_pretrained(mod)
+    #feature_extractor2 = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
     model2 = AutoModelForImageClassification.from_pretrained(mod)
     input = feature_extractor2(image, return_tensors="pt")
     with torch.no_grad():