Update app.py

app.py

@@ -31,6 +31,7 @@ model.generation_config.eos_token_id = processor.tokenizer.eos_token_id
 @spaces.GPU(duration=120)
 def krypton(input, history):
     if input["files"]:
+        print("found the image\n")
         image = input["files"][-1]["path"] if isinstance(input["files"][-1], dict) else input["files"][-1]
     else:
         image = None
@@ -43,10 +44,12 @@ def krypton(input, history):
         return
     
     prompt = f"user\n\n<image>\n{input['text']}\nassistant\n\n"
+    print("made the prompt")
     image = Image.open(image)
     inputs = processor(prompt, images=image, return_tensors='pt').to(0, torch.float16)
     
     # Streamer
+    print('About to init streamer')
     streamer = TextIteratorStreamer(processor.tokenizer, skip_special_tokens=False, skip_prompt=True)
     
     # Generation kwargs
@@ -59,6 +62,7 @@ def krypton(input, history):
     )
     
     thread = threading.Thread(target=model.generate, kwargs=generation_kwargs)
+    print('threader about to start')
     thread.start()
     
     buffer = ""