Update app.py

app.py

@@ -1,29 +1,12 @@
 import random
-import time
-from io import BytesIO
-from urllib.parse import quote
 
-from datasets import load_dataset
 import gradio as gr
-import requests
-from PIL import Image
+from datasets import load_dataset
+
+from client import Inference
 
 MAX_SEED = 2 ** 31 - 1
-session = requests.Session()
-
-def _fetch(*args, retries=5, **kwargs):
-    for i in range(retries + 1):
-        try:
-            response = session.get(*args, **kwargs)
-            response.raise_for_status()
-            return response.content
-        except requests.RequestException:
-            if response.status_code in {429, 500, 502, 503, 504}:
-                delay = min(2 ** i, 60)
-                time.sleep(delay)
-            else:
-                return None
-    return None
+client = Inference()
 
 def load_examples():
     ds = load_dataset("k-mktr/improved-flux-prompts", split="train")
@@ -38,20 +21,15 @@ def generate(
 ):
     if seed is None or randomize_seed:
         seed = random.randint(0, MAX_SEED)
-    url = f"https://pollinations.ai/p/{quote(prompt)}?nologo=true&private=true"
-    params = dict(
+    return client(
         prompt=prompt,
         seed=seed,
         width=width,
         height=height
     )
-    image = _fetch(url, params=params)
-    if image:
-        return Image.open(BytesIO(image))
-    return None
 
 with gr.Blocks(
-    title="Text-to-Image",
+    title="🎨 Text-to-Image",
     css="footer {display: none !important}",
     theme=gr.themes.Base(
         primary_hue="red",
@@ -83,7 +61,7 @@ with gr.Blocks(
             inputs=[prompt],
             examples_per_page=2
         )
-        randomize_btn = gr.Button("🎲 Randomize")
+        randomize_btn = gr.Button("See more")
             
     with gr.Tab("⚙️ Settings"):
         with gr.Row():
@@ -119,7 +97,7 @@ with gr.Blocks(
             format="jpeg",
             interactive=False
         )
-        clear_btn = gr.Button("🗑️ Clear")
+        clear_btn = gr.Button("Clear")
 
     with gr.Accordion("ℹ️ About", open=False):
         gr.Markdown(f"""