Spaces:
Runtime error
Runtime error
update app
Browse files- app.py +1 -0
- requirements.txt +2 -1
app.py
CHANGED
|
@@ -19,6 +19,7 @@ r_model = T5ForMultimodalGeneration.from_pretrained(rationale_model_dir, patch_s
|
|
| 19 |
|
| 20 |
def inference_chat(input_image,input_text):
|
| 21 |
with torch.no_grad():
|
|
|
|
| 22 |
img = Image.open(input_image).convert("RGB")
|
| 23 |
input = transform(img).unsqueeze(0)
|
| 24 |
out = vit_model.forward_features(input)
|
|
|
|
| 19 |
|
| 20 |
def inference_chat(input_image,input_text):
|
| 21 |
with torch.no_grad():
|
| 22 |
+
print(input_image)
|
| 23 |
img = Image.open(input_image).convert("RGB")
|
| 24 |
input = transform(img).unsqueeze(0)
|
| 25 |
out = vit_model.forward_features(input)
|
requirements.txt
CHANGED
|
@@ -1,5 +1,6 @@
|
|
|
|
|
| 1 |
git+https://github.com/huggingface/transformers.git
|
| 2 |
torch
|
| 3 |
torchvision
|
| 4 |
sentencepiece
|
| 5 |
-
numpy
|
|
|
|
| 1 |
+
huggingface-hub>=0.4.0
|
| 2 |
git+https://github.com/huggingface/transformers.git
|
| 3 |
torch
|
| 4 |
torchvision
|
| 5 |
sentencepiece
|
| 6 |
+
numpy
|