Update app.py

app.py

@@ -30,8 +30,6 @@ from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
 
-HEADER = """FRAME AI"""
-
 
 if torch.cuda.is_available():
     device = "cuda:0"
@@ -134,8 +132,8 @@ def generate_image_from_text(pos_prompt, seed):
                             'negativeText': neg_prompt},
       'imageGenerationConfig': {"cfgScale":8,
                                 "seed":int(seed),
-                                "width":512,
-                                "height":512,
+                                "width":768,
+                                "height":768,
                                 "numberOfImages":1
                                 }
   }
@@ -144,16 +142,8 @@ def generate_image_from_text(pos_prompt, seed):
   response_body = json.loads(response.get('body').read())
   base64_image_data = base64.b64decode(response_body['images'][0])
 
-  image = Image.open(io.BytesIO(base64_image_data))
-
-
-  # webp_image_io = io.BytesIO()
-  # image = image.convert('RGB')  # Convert to RGB for WebP compatibility
-  image.save(image, format='WEBP')
-
-  # webp_image_io.seek(0)
+  return Image.open(io.BytesIO(base64_image_data))
 
-  return Image.open(image)
 
 def check_input_image(input_image):
     if input_image is None: