Spaces:
Sleeping
Sleeping
File size: 2,549 Bytes
90bece4 33c27ec 90bece4 7521b47 90bece4 30b12a2 7521b47 9184aae 5d123f9 90bece4 7521b47 3e7d217 90bece4 30b12a2 3e7d217 30b12a2 7521b47 30b12a2 90bece4 5c84876 90bece4 5c84876 7521b47 5c84876 3e7d217 90bece4 5d123f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
# main.py
import requests
import os
import warnings
import io
import getpass
from PIL import Image
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, download_path):
headers = {'x-api-key': clipdrop_api_key}
body_params = {'prompt': (None, text_prompt, 'text/plain')}
response = requests.post('https://clipdrop-api.co/text-to-image/v1',
files=body_params,
headers=headers)
if response.status_code != 200:
print(f"Request failed with status code {response.status_code}")
return None
with open('generated_image.png', 'wb') as f:
f.write(response.content)
os.environ['STABILITY_HOST'] = 'grpc.stability.ai:443'
os.environ['STABILITY_KEY'] = stability_api_key
stability_api = client.StabilityInference(
key=os.environ['STABILITY_KEY'],
upscale_engine="esrgan-v1-x2plus",
verbose=True,
)
max_pixels = 1048576
img = Image.open('generated_image.png')
width, height = img.size
if width * height > max_pixels:
scale_factor = (max_pixels / (width * height))**0.5
new_width = int(width * scale_factor)
new_height = int(height * scale_factor)
img = img.resize((new_width, new_height))
answers = stability_api.upscale(init_image=img)
for resp in answers:
for artifact in resp.artifacts:
if artifact.finish_reason == generation.FILTER:
warnings.warn(
"Your request activated the API's safety filters and could not be processed."
"Please submit a different image and try again.")
if artifact.type == generation.ARTIFACT_IMAGE:
upscaled_img = Image.open(io.BytesIO(artifact.binary))
upscaled_img.save("upscaled_image.png")
os.environ['REPLICATE_API_TOKEN'] = replicate_api_token
Image.MAX_IMAGE_PIXELS = None
with open("upscaled_image.png", "rb") as img_file:
output = replicate.run(
"tencentarc/gfpgan:9283608cc6b7be6b65a8e44983db012355fde4132009bf99d976b2f0896856a3",
input={"img": img_file, "version": "v1.4", "scale": 16}
)
response = requests.get(output)
final_img = Image.open(io.BytesIO(response.content))
final_img.save(os.path.join(download_path, "gfpgan_upscaled_image.png"))
|