AI-or-Not-v2

Running

Omnibus commited on Oct 13, 2023

Commit

d140aac

1 Parent(s): b936e5d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -35,6 +35,8 @@ def aiornot(image,mod_choose):
     input = feature_extractor(image, return_tensors="pt")
     with torch.no_grad():
       outputs = model(**input)
       logits = outputs.logits
       print (logits)
     prediction = logits.argmax(-1).item()

     input = feature_extractor(image, return_tensors="pt")
     with torch.no_grad():
       outputs = model(**input)
+      print(outputs)
+      print(dir(outputs))
       logits = outputs.logits
       print (logits)
     prediction = logits.argmax(-1).item()