Update app.py

app.py

@@ -52,7 +52,7 @@ def aiornot0(image):
 def aiornot1(image):    
     labels = ["Real", "AI"]
     mod=models[1]
-    feature_extractor1 = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
+    feature_extractor1 = AutoFeatureExtractor.from_pretrained(mod)
     model1 = AutoModelForImageClassification.from_pretrained(mod)
     input = feature_extractor1(image, return_tensors="pt")
     with torch.no_grad():
@@ -76,7 +76,7 @@ def aiornot1(image):
 def aiornot2(image):    
     labels = ["AI", "Real"]
     mod=models[2]
-    feature_extractor2 = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
+    feature_extractor2 = AutoFeatureExtractor.from_pretrained(mod)
     model2 = AutoModelForImageClassification.from_pretrained(mod)
     input = feature_extractor2(image, return_tensors="pt")
     with torch.no_grad():
@@ -101,7 +101,7 @@ def aiornot2(image):
 def aiornot3(image):    
     labels = ["Real", "AI"]
     mod=models[3]
-    feature_extractor3 = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
+    feature_extractor3 = AutoFeatureExtractor.from_pretrained(mod)
     model3 = AutoModelForImageClassification.from_pretrained(mod)
     input = feature_extractor3(image, return_tensors="pt")
     with torch.no_grad():