Update app.py

app.py

@@ -212,7 +212,7 @@ EXAMPLES = [
 ]
 
 @spaces.GPU()
-def simple_chat(message, history: list, temperature: float = 0.8, max_length: int = 4096, top_p: float = 1, top_k: int = 10, penalty: float = 1.0):
+def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p: float = 1, top_k: int = 10, penalty: float = 1.0):
     
     model = AutoModelForCausalLM.from_pretrained(
         MODEL_ID,
@@ -222,7 +222,7 @@ def simple_chat(message, history: list, temperature: float = 0.8, max_length: in
     )
         
     print(f'message is - {message}')
-    print(f'history is - {history}')
+
     conversation = []
     prompt_files = []
     if message["files"]:
@@ -233,13 +233,7 @@ def simple_chat(message, history: list, temperature: float = 0.8, max_length: in
             format_msg = contents + "\n\n\n" + "{} files uploaded.\n" + message['text']
             conversation.append({"role": "user", "content": format_msg})
     else:
-        if len(history) == 0:
-            # raise gr.Error("Please upload an image first.")
-            contents = None
-            conversation.append({"role": "user", "content": message['text']})
-        else:
-            # image = Image.open(history[0][0][0])
-            for prompt, answer in history:
+        for prompt, answer in history:
                 if answer is None:
                     prompt_files.append(prompt[0])
                     conversation.extend([{"role": "user", "content": ""}, {"role": "assistant", "content": ""}])
@@ -257,6 +251,7 @@ def simple_chat(message, history: list, temperature: float = 0.8, max_length: in
             elif choice == "doc":
                 format_msg = contents + "\n\n\n" + "{} files uploaded.\n" + message['text']
                 conversation.append({"role": "user", "content": format_msg})
+            
     print(f"Conversation is -\n{conversation}")
 
     input_ids = tokenizer.apply_chat_template(conversation, tokenize=True, add_generation_prompt=True,