File size: 609 Bytes
d61f523 |
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 |
import gradio as gr
import requests
import random
import sys
import os
def download(url):
filename = str(random.randint(111111111, 999999999)) + ".png"
file_path = os.path.join(os.path.dirname(__file__), filename)
response = requests.get(url)
if response.status_code == 200:
with open(file_path, "wb") as f:
f.write(response.content)
return(file_path)
def process(prompt):
download_file("https://image.pollinations.ai/prompt/" + text + ('.' * random.randint(0, 100)))
iface = gr.Interface(fn=process, inputs="text", outputs="image")
iface.launch() |