han1997 commited on
Commit
d9a4193
·
verified ·
1 Parent(s): 0d355d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -60,6 +60,8 @@ def bot_streaming(message, history, temperature, top_k, max_new_tokens):
60
 
61
  if image is not None:
62
  image = Image.open(image).convert("RGB")
 
 
63
 
64
  prompt_builder.add_turn(role="human", message=message['text'])
65
  prompt_text = prompt_builder.get_prompt()
@@ -76,9 +78,6 @@ def bot_streaming(message, history, temperature, top_k, max_new_tokens):
76
  }
77
 
78
  # Generate from the VLM
79
- # buffer = vlm.generate(
80
- # **generation_kwargs
81
- # )
82
  thread = Thread(target=vlm.generate, kwargs=generation_kwargs)
83
  thread.start()
84
 
 
60
 
61
  if image is not None:
62
  image = Image.open(image).convert("RGB")
63
+
64
+ print(image)
65
 
66
  prompt_builder.add_turn(role="human", message=message['text'])
67
  prompt_text = prompt_builder.get_prompt()
 
78
  }
79
 
80
  # Generate from the VLM
 
 
 
81
  thread = Thread(target=vlm.generate, kwargs=generation_kwargs)
82
  thread.start()
83