yejunliang23 commited on
Commit
7499b16
Β·
unverified Β·
1 Parent(s): 40ce34a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -42,14 +42,12 @@ def predict(_chatbot, task_history):
42
  messages.append({'role': 'user', 'content': content})
43
  messages.append({'role': 'assistant', 'content': [{'text': a}]})
44
  content = []
45
- print(messages)
46
  messages.pop()
47
- print(messages)
48
 
49
  messages = _transform_messages(messages)
50
- print(messages)
51
  text = processor.apply_chat_template(
52
  messages, tokenize=False, add_generation_prompt=True)
 
53
  image_inputs, video_inputs = process_vision_info(messages)
54
  inputs = processor(text=[text], images=image_inputs,
55
  videos=video_inputs, padding=True, return_tensors='pt')
 
42
  messages.append({'role': 'user', 'content': content})
43
  messages.append({'role': 'assistant', 'content': [{'text': a}]})
44
  content = []
 
45
  messages.pop()
 
46
 
47
  messages = _transform_messages(messages)
 
48
  text = processor.apply_chat_template(
49
  messages, tokenize=False, add_generation_prompt=True)
50
+ print(text)
51
  image_inputs, video_inputs = process_vision_info(messages)
52
  inputs = processor(text=[text], images=image_inputs,
53
  videos=video_inputs, padding=True, return_tensors='pt')