Omnibus commited on
Commit
d140aac
1 Parent(s): b936e5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -35,6 +35,8 @@ def aiornot(image,mod_choose):
35
  input = feature_extractor(image, return_tensors="pt")
36
  with torch.no_grad():
37
  outputs = model(**input)
 
 
38
  logits = outputs.logits
39
  print (logits)
40
  prediction = logits.argmax(-1).item()
 
35
  input = feature_extractor(image, return_tensors="pt")
36
  with torch.no_grad():
37
  outputs = model(**input)
38
+ print(outputs)
39
+ print(dir(outputs))
40
  logits = outputs.logits
41
  print (logits)
42
  prediction = logits.argmax(-1).item()