Singularity666 commited on
Commit
5d123f9
·
1 Parent(s): 5f0e6f8

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -4
main.py CHANGED
@@ -11,7 +11,7 @@ from stability_sdk import client
11
  import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation
12
  import replicate
13
 
14
- def generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key, replicate_api_token):
15
 
16
  headers = {'x-api-key': clipdrop_api_key}
17
  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,
69
 
70
  response = requests.get(output)
71
  final_img = Image.open(io.BytesIO(response.content))
72
- final_img.save("gfpgan_upscaled_image.png")
73
-
74
- return final_img
 
11
  import stability_sdk.interfaces.gooseai.generation.generation_pb2 as generation
12
  import replicate
13
 
14
+ def generate_and_upscale_image(text_prompt, clipdrop_api_key, stability_api_key, replicate_api_token, download_path):
15
 
16
  headers = {'x-api-key': clipdrop_api_key}
17
  body_params = {'prompt': (None, text_prompt, 'text/plain')}
 
69
 
70
  response = requests.get(output)
71
  final_img = Image.open(io.BytesIO(response.content))
72
+ final_img.save(os.path.join(download_path, "gfpgan_upscaled_image.png"))