Update app.py

app.py

@@ -213,7 +213,6 @@ EXAMPLES = [
 
 @spaces.GPU()
 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,
         torch_dtype=torch.bfloat16,
@@ -221,20 +220,9 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
         trust_remote_code=True
     )
         
-    print(f'message is - {message}')
+    print(f'Message is - {message}')
 
     conversation = []
-    prompt_files = []
-    # if message["files"]:
-    #     choice, contents = mode_load(message["files"][-1])
-    #     if choice == "image":
-    #         conversation.append({"role": "user", "image": contents, "content": message['text']})
-    #     elif choice == "doc":
-    #         format_msg = contents + "\n\n\n" + "{} files uploaded.\n" + message['text']
-    #         conversation.append({"role": "user", "content": format_msg})
-    # else:
-    #     choice = ""
-    #     conversation.append({"role": "user", "image": "", "content": message['text']})
 
     if message["file_content"]:
         choice, contents = mode_load(message["file_content"])
@@ -248,11 +236,9 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
 
     print(f"Conversation is -\n{conversation}")
 
-            
-    print(f"Conversation is -\n{conversation}")
-
     input_ids = tokenizer.apply_chat_template(conversation, tokenize=True, add_generation_prompt=True,
                                               return_tensors="pt", return_dict=True).to(model.device)
+    
     streamer = TextIteratorStreamer(tokenizer, timeout=60.0, skip_prompt=True, skip_special_tokens=True)
 
     generate_kwargs = dict(
@@ -274,17 +260,14 @@ def simple_chat(message, temperature: float = 0.8, max_length: int = 4096, top_p
         for new_text in streamer:
             buffer += new_text
             yield buffer
-        print(" ")
+        
         print("---------")
         print("Text: ")
-        print("  ")
         print(buffer)
-        print(" ")
         print("---------")
 
 
 
-
 # @spaces.GPU()
 # def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096, top_p: float = 1, top_k: int = 10, penalty: float = 1.0):
 #     try: