Update app.py

app.py

@@ -155,9 +155,7 @@ def process_document_stream(model_name: str, image: Image.Image, max_new_tokens:
     if image is None:
         yield "Please upload an image.", "Please upload an image."
         return
-
-    text_prompt = prompt_input
-
+        
     if model_name == "Camel-Doc-OCR-080125":
         processor, model = processor_m, model_m
     elif model_name == "Megalodon-OCR-Sync-0713":
@@ -172,7 +170,7 @@ def process_document_stream(model_name: str, image: Image.Image, max_new_tokens:
         yield "Invalid model selected.", "Invalid model selected."
         return
 
-    messages = [{"role": "user", "content": [{"type": "image", "image": image}, {"type": "text", "text": text_prompt}]}]
+    messages = [{"role": "user", "content": [{"type": "image", "image": image}, {"type": "text", "text": prompt_input}]}]
     prompt_full = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
     inputs = processor(text=[prompt_full], images=[image], return_tensors="pt", padding=True, truncation=True, max_length=MAX_INPUT_TOKEN_LENGTH).to(device)
     streamer = TextIteratorStreamer(processor, skip_prompt=True, skip_special_tokens=True)