Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -31,7 +31,8 @@ model = AutoModelForVision2Seq.from_pretrained(model_id, torch_dtype="auto", dev
|
|
31 |
@spaces.GPU
|
32 |
def stream_chat(message: Dict[str, str], history: list):
|
33 |
# {'text': 'what is this', 'files': ['image-xxx.jpg']}
|
34 |
-
|
|
|
35 |
|
36 |
image = Image.open(message["files"][0])
|
37 |
pixel_values = processor(images=[image], return_tensors="pt").to(model.device)["pixel_values"]
|
|
|
31 |
@spaces.GPU
|
32 |
def stream_chat(message: Dict[str, str], history: list):
|
33 |
# {'text': 'what is this', 'files': ['image-xxx.jpg']}
|
34 |
+
print(message)
|
35 |
+
print(history)
|
36 |
|
37 |
image = Image.open(message["files"][0])
|
38 |
pixel_values = processor(images=[image], return_tensors="pt").to(model.device)["pixel_values"]
|