Update app.py

app.py

@@ -30,21 +30,22 @@ model.generation_config.eos_token_id = processor.tokenizer.eos_token_id
 
 @spaces.GPU(duration=120)
 def krypton(input, history):
+    print(f"Input: {input}")  # Debug input
+    print(f"History: {history}")  # Debug history
+
     if input["files"]:
-        print("found the image\n")
+        print("Found the image\n")
         image_path = input["files"][-1]["path"] if isinstance(input["files"][-1], dict) else input["files"][-1]
+        print(f"Image path: {image_path}")
     else:
         image_path = None
         for hist in history:
             if isinstance(hist[0], tuple):
-                image = hist[0][0]
+                image_path = hist[0][0]
     
     if not image_path:
         gr.Error("You need to upload an image for Krypton to work.")
         return
-    
-    prompt = f"user\n\n<image>\n{input['text']}\nassistant\n\n"
-    print("Made the prompt")
 
     try:
         image = Image.open(image_path)
@@ -54,6 +55,10 @@ def krypton(input, history):
         gr.Error("Failed to open the image.")
         return
 
+    # Adding more context to the prompt with a placeholder for the image
+    prompt = f"user: Here is an image and a question about it.\n<image>{input['text']}\nassistant: "
+    print("Made the prompt")
+
     try:
         inputs = processor(prompt, images=image, return_tensors='pt').to('cuda', torch.float16)
         print(f"Processed inputs: {inputs}")
@@ -62,7 +67,6 @@ def krypton(input, history):
         gr.Error("Failed to process the inputs.")
         return
     
-    
     # Streamer
     print('About to init streamer')
     streamer = TextIteratorStreamer(processor.tokenizer, skip_special_tokens=False, skip_prompt=True)
@@ -77,7 +81,7 @@ def krypton(input, history):
     )
     
     thread = threading.Thread(target=model.generate, kwargs=generation_kwargs)
-    print('threader about to start')
+    print('Thread about to start')
     thread.start()
     
     buffer = ""