Update app.py

app.py

@@ -4,6 +4,7 @@ import os
 import base64
 from PIL import Image
 import io
+import json
 
 def resize_image(image_path, max_size=(800, 800), quality=85):
     with Image.open(image_path) as img:
@@ -60,11 +61,15 @@ def call_fuyu_8b_api(image_path, content, temperature=0.2, top_p=0.7, max_tokens
         for line in response.iter_lines():
             if line:
                 decoded_line = line.decode("utf-8")
-                if "content" in decoded_line:
-                    response_text += decoded_line.split('"content":"')[1].split('","finish_reason')[0]
+                try:
+                    # Parse each line as JSON and extract the "content" part
+                    json_line = json.loads(decoded_line)
+                    content_part = json_line.get("delta", {}).get("content", "")
+                    response_text += content_part
+                except json.JSONDecodeError:
+                    print("Error decoding JSON from response line.")
         
         return response_text
-
 content_input = gr.Textbox(lines=2, placeholder="Enter your content here...", label="Content")
 image_input = gr.Image(type="filepath", label="Upload Image")
 temperature_input = gr.Slider(minimum=0, maximum=1, step=0.01, value=0.2, label="Temperature")