Update app.py

app.py

@@ -49,7 +49,8 @@ eos_token_id=processor.tokenizer.eos_token_id
 
 @spaces.GPU(queue=False)
 def stream_chat(message, history: list, temperature: float, max_new_tokens: int):
-    print(message)
+    print(f'message is - {message}')
+    print(f'history is - {history}')
     conversation = []
     for prompt, answer in history:
         conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
@@ -60,10 +61,11 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
     else:
         if len(history) == 0:
             raise gr.Error("Please upload an image first.")
-        image = None
+            image = None
+        elif len(history):
+            image = history        
         conversation.append({"role": "user", "content": message['text']})
-    print(conversation)
-    
+    print(f"Conversation is -\n{conversation}")
     inputs = processor.tokenizer.apply_chat_template(conversation, tokenize=False, add_generation_prompt=True)
     inputs_ids = processor(inputs, image, return_tensors="pt").to(0)
     streamer = TextIteratorStreamer(processor, **{"skip_special_tokens": True, "skip_prompt": True, 'clean_up_tokenization_spaces':False,})