Omnibus commited on
Commit
9b288a0
1 Parent(s): c7c6c17

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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(mod)
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():