hiyouga commited on
Commit
4f26d86
·
verified ·
1 Parent(s): bca5e76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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"]