fromated the try and except blocks for krypton

app.py

@@ -26,11 +26,10 @@ model.to('cuda')
 processor = AutoProcessor.from_pretrained(model_id)
 
 # Confirming and setting the eos_token_id (if necessary)
-# model.generation_config.eos_token_id = processor.tokenizer.eos_token_id
 model.generation_config.eos_token_id = 128009
 
 @spaces.GPU(duration=120)
-def bot_streaming(message, history):
+def krypton(message, history):
     print(message)
     if message["files"]:
         # message["files"][-1] is a Dict or just a string
@@ -47,13 +46,12 @@ def bot_streaming(message, history):
     try:
         if image is None:
             # Handle the case where image is None
-            gr.Error("You need to upload an image for LLaVA to work.")
+            gr.Error("Please upload an image so kypton can understand.")
     except NameError:
         # Handle the case where 'image' is not defined at all
-        gr.Error("You need to upload an image for LLaVA to work.")
+        gr.Error("Upload an image so krypton can work.")
 
     prompt = f"<|start_header_id|>user<|end_header_id|>\n\n<image>\n{message['text']}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
-    # print(f"prompt: {prompt}")
     image = Image.open(image)
     inputs = processor(prompt, image, return_tensors='pt').to(0, torch.float16)
 
@@ -64,7 +62,6 @@ def bot_streaming(message, history):
     thread.start()
 
     text_prompt = f"<|start_header_id|>user<|end_header_id|>\n\n{message['text']}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
-    # print(f"text_prompt: {text_prompt}")
 
     buffer = ""
     time.sleep(0.5)
@@ -73,12 +70,8 @@ def bot_streaming(message, history):
         if "<|eot_id|>" in new_text:
             new_text = new_text.split("<|eot_id|>")[0]
         buffer += new_text
-
-        # generated_text_without_prompt = buffer[len(text_prompt):]
         generated_text_without_prompt = buffer
-        # print(generated_text_without_prompt)
         time.sleep(0.06)
-        # print(f"new_text: {generated_text_without_prompt}")
         yield generated_text_without_prompt
 
 
@@ -88,7 +81,7 @@ chat_input = gr.MultimodalTextbox(interactive=True, file_types=["image"], placeh
 with gr.Blocks(fill_height=True) as demo:
     gr.Markdown(DESCRIPTION)
     gr.ChatInterface(
-        fn=bot_streaming,
+        fn=kypton,
         chatbot=chatbot,
         fill_height=True,
         multimodal=True,