Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -115,7 +115,7 @@ def aiornot1(image):
|
|
115 |
def aiornot2(image):
|
116 |
labels = ["AI", "Real"]
|
117 |
mod=models[2]
|
118 |
-
feature_extractor2 = AutoFeatureExtractor.from_pretrained(
|
119 |
model2 = AutoModelForImageClassification.from_pretrained(mod)
|
120 |
input = feature_extractor2(image, return_tensors="pt")
|
121 |
with torch.no_grad():
|
|
|
115 |
def aiornot2(image):
|
116 |
labels = ["AI", "Real"]
|
117 |
mod=models[2]
|
118 |
+
feature_extractor2 = AutoFeatureExtractor.from_pretrained("microsoft/resnet-50")
|
119 |
model2 = AutoModelForImageClassification.from_pretrained(mod)
|
120 |
input = feature_extractor2(image, return_tensors="pt")
|
121 |
with torch.no_grad():
|