Update app.py

app.py

@@ -58,19 +58,13 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
         else:
             image = Image.open(history[0][0][0])
             for prompt, answer in history:
- #               if answer is None:
- #                   conversation.extend([{"role": "user", "content":"<|image_1|>"},{"role": "assistant", "content": ""}])
- #               else:
-                    conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
+                conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
             conversation.append({"role": "user", "content": message['text']})    
     print(f"Conversation is -\n{conversation}")
 
-  #  streamer = TextIteratorStreamer(tokenizer, **{"skip_special_tokens": True, "skip_prompt": True, 'clean_up_tokenization_spaces':False,}) 
-
     generate_kwargs = dict(
         image=image,
         msgs=conversation,
-#        streamer=streamer,
         max_new_tokens=max_new_tokens,
         temperature=temperature,
         sampling=True,
@@ -78,16 +72,7 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
     )
     if temperature == 0:
         generate_kwargs["sampling"] = False
-    
-"""
-    thread = Thread(target=model.chat, kwargs=generate_kwargs)
-    thread.start()
 
-    buffer = ""
-    for new_text in streamer:
-        buffer += new_text
-        yield buffer
-"""
     response = model.chat(**generate_kwargs)
     return response