please
Browse files- app.py +1 -2
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -10,5 +10,4 @@ image = Image.open(requests.get(url, stream=True).raw)
|
|
| 10 |
|
| 11 |
inputs = feature_extractor(images=image, return_tensors="pt")
|
| 12 |
outputs = model(**inputs)
|
| 13 |
-
logits = outputs.logits
|
| 14 |
-
# shape (batch_size, num_labels, height/4, width/4)
|
|
|
|
| 10 |
|
| 11 |
inputs = feature_extractor(images=image, return_tensors="pt")
|
| 12 |
outputs = model(**inputs)
|
| 13 |
+
logits = outputs.logits # shape (batch_size, num_labels, height/4, width/4)
|
|
|
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
torch
|
| 2 |
transformers
|
|
|
|
| 3 |
tensorflow
|
| 4 |
numpy
|
| 5 |
Image
|
|
|
|
| 1 |
torch
|
| 2 |
transformers
|
| 3 |
+
tensorRT
|
| 4 |
tensorflow
|
| 5 |
numpy
|
| 6 |
Image
|