Update app.py

app.py

@@ -3,6 +3,8 @@ import os
 from together import Together
 import base64
 from io import BytesIO
+from PIL import Image
+import numpy as np
 
 # Initialize the Together client
 api_key = os.environ.get('TOGETHER_API_KEY')
@@ -19,6 +21,10 @@ def generate_gradio_app(image):
         return "Error: TOGETHER_API_KEY not set or client initialization failed. Please check your API key."
 
     try:
+        # Convert numpy array to PIL Image
+        if isinstance(image, np.ndarray):
+            image = Image.fromarray(image.astype('uint8'), 'RGB')
+        
         # Convert the image to base64
         buffered = BytesIO()
         image.save(buffered, format="PNG")