amirgame197's picture
Update app.py
97b7aea verified
raw
history blame contribute delete
759 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 + "?model=flux-realism&width=2048&height=2048&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()