new

app.py

@@ -284,7 +284,7 @@ def image_to_base64(image_path):
     """Converts an image file to a base64-encoded string."""
     with open(image_path, "rb") as img_file:
         return base64.b64encode(img_file.read()).decode('utf-8')
-
+@spaces.GPU
 def main():
 
     torch.set_grad_enabled(False)