Update app.py
Browse files
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()
|