HuanjinYao commited on
Commit
baaa59a
1 Parent(s): a637de1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -82,6 +82,7 @@ def bot_streaming(message, history):
82
  conv.append_message(conv.roles[0], user)
83
  conv.append_message(conv.roles[1], None)
84
  prompt = conv.get_prompt()
 
85
 
86
  image = Image.open(image).convert('RGB')
87
  image_tensor = process_images([image], image_processor, model.config)[0]
@@ -99,6 +100,7 @@ def bot_streaming(message, history):
99
  buffer = ""
100
  # time.sleep(0.5)
101
  for new_text in streamer:
 
102
  if "<|eot_id|>" in new_text:
103
  new_text = new_text.split("<|eot_id|>")[0]
104
  buffer += new_text
 
82
  conv.append_message(conv.roles[0], user)
83
  conv.append_message(conv.roles[1], None)
84
  prompt = conv.get_prompt()
85
+ print(prompt)
86
 
87
  image = Image.open(image).convert('RGB')
88
  image_tensor = process_images([image], image_processor, model.config)[0]
 
100
  buffer = ""
101
  # time.sleep(0.5)
102
  for new_text in streamer:
103
+ print('new_text', new_text)
104
  if "<|eot_id|>" in new_text:
105
  new_text = new_text.split("<|eot_id|>")[0]
106
  buffer += new_text