Update app.py

app.py

@@ -263,21 +263,18 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
             generated_ids = model.generate(input_ids['input_ids'], **generate_kwargs)
             generated_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)
 
-        text_original = message['text'].strip()
-        generated_text_cleaned = generated_text.replace(text_original, "").strip()
+        # text_original = message['text'].strip()
+        # generated_text_cleaned = generated_text.replace(text_original, "").strip()
 
-
-        print("---------")
-        print("IDs: ")
+)
         print(" ")
-        print(generated_ids[0])
         print("---------")
         print("Text: ")
         print("  ")
-        print(generated_text_cleaned)
+        print(generated_text)
 
 
-        return PlainTextResponse(generated_text_cleaned)
+        return PlainTextResponse(generated_text)
     except Exception as e:
         return PlainTextResponse(f"Error: {str(e)}")