Omnibus commited on
Commit
cafb60e
·
1 Parent(s): 88cf8f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -44,7 +44,7 @@ def aiornot1(image):
44
  model1 = AutoModelForImageClassification.from_pretrained(mod)
45
  input = feature_extractor1(image, return_tensors="pt")
46
  with torch.no_grad():
47
- outputs = model1(**input)
48
  logits = outputs.logits
49
  probability = softmax(logits)
50
  px = pd.DataFrame(probability.numpy())
@@ -66,7 +66,7 @@ def aiornot2(image):
66
  model2 = AutoModelForImageClassification.from_pretrained(mod)
67
  input = feature_extractor2(image, return_tensors="pt")
68
  with torch.no_grad():
69
- outputs = model2(**input)
70
  logits = outputs.logits
71
  probability = softmax(logits)
72
  px = pd.DataFrame(probability.numpy())
 
44
  model1 = AutoModelForImageClassification.from_pretrained(mod)
45
  input = feature_extractor1(image, return_tensors="pt")
46
  with torch.no_grad():
47
+ outputs = model1(**input)
48
  logits = outputs.logits
49
  probability = softmax(logits)
50
  px = pd.DataFrame(probability.numpy())
 
66
  model2 = AutoModelForImageClassification.from_pretrained(mod)
67
  input = feature_extractor2(image, return_tensors="pt")
68
  with torch.no_grad():
69
+ outputs = model2(**input)
70
  logits = outputs.logits
71
  probability = softmax(logits)
72
  px = pd.DataFrame(probability.numpy())