amirgame197's picture
Update app.py
e475598 verified
raw
history blame contribute delete
741 Bytes
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 + "&width=1536&height=1536&nologo=true&seed=" + str(random.randint(0, 999999999)))
if response.status_code == 200:
with open(file_path, "wb") as f:
f.write(response.content)
return(file_path)
title = "Pollinations Image Generator"
description = "Pollinations API + Randomizer"
iface = gr.Interface(fn=process, inputs="text", outputs="image", title=title, description=description)
iface.launch()