Update app.py

app.py

@@ -32,22 +32,36 @@ model.generation_config.eos_token_id = processor.tokenizer.eos_token_id
 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]
+        image_path = input["files"][-1]["path"] if isinstance(input["files"][-1], dict) else input["files"][-1]
     else:
-        image = None
+        image_path = None
         for hist in history:
             if isinstance(hist[0], tuple):
                 image = hist[0][0]
     
-    if not image:
+    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")
-    image = Image.open(image)
-    print(f"image open: {image}")
-    inputs = processor(prompt, images=image, return_tensors='pt').to(0, torch.float16)
+    print("Made the prompt")
+
+    try:
+        image = Image.open(image_path)
+        print(f"Image open: {image}")
+    except Exception as e:
+        print(f"Error opening image: {e}")
+        gr.Error("Failed to open the image.")
+        return
+
+    try:
+        inputs = processor(prompt, images=image, return_tensors='pt').to('cuda', torch.float16)
+        print(f"Processed inputs: {inputs}")
+    except Exception as e:
+        print(f"Error processing inputs: {e}")
+        gr.Error("Failed to process the inputs.")
+        return
+    
     
     # Streamer
     print('About to init streamer')