Tonic commited on
Commit
75bc022
1 Parent(s): 3897eaf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -42,7 +42,7 @@ class ChatBot:
42
  query_elements.append({'text': text_query})
43
 
44
  query = self.tokenizer.from_list_format(query_elements)
45
- response, self.history = self.model.chat(self.tokenizer, query=query, history=self.history)
46
  return response
47
 
48
  def draw_boxes(self, response, image_path):
 
42
  query_elements.append({'text': text_query})
43
 
44
  query = self.tokenizer.from_list_format(query_elements)
45
+ response, self.history = self.model.chat(self.tokenizer, query=query, history=self.history, max_new_tokens = 1200)
46
  return response
47
 
48
  def draw_boxes(self, response, image_path):