Spaces:
Running
Running
import gradio as gr | |
import requests | |
import random | |
import sys | |
import os | |
def process(prompt): | |
filename = str(random.randint(111111111, 999999999)) + ".png" | |
file_path = os.path.join(os.path.dirname(__file__), filename) | |
response = requests.get("https://image.pollinations.ai/prompt/" + prompt + ('.' * random.randint(0, 100))) | |
if response.status_code == 200: | |
with open(file_path, "wb") as f: | |
f.write(response.content) | |
return(file_path) | |
iface = gr.Interface(fn=process, inputs="text", outputs="image") | |
iface.launch() |