Update main.py

main.py

@@ -11,7 +11,7 @@ from stability_sdk import client
 import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation
 import replicate
 
-def generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key, replicate_api_token):
+def generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key, replicate_api_token, download_path):
 
     headers = {'x-api-key': clipdrop_api_key}
     body_params = {'prompt': (None, text_prompt, 'text/plain')}
@@ -69,6 +69,4 @@ def generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key,
 
     response = requests.get(output)
     final_img = Image.open(io.BytesIO(response.content))
-    final_img.save("gfpgan_upscaled_image.png")
-
-    return final_img
+    final_img.save(os.path.join(download_path, "gfpgan_upscaled_image.png"))