Update app.py

app.py

@@ -1,7 +1,9 @@
 import gradio as gr
 import cv2
 import numpy as np
+import os
 from datetime import datetime
+from huggingface_hub import HfInferenceEndpoint
 import random
 
 #----------Start of theme----------
@@ -38,6 +40,12 @@ theme = gr.themes.Soft(
 )
 #----------End of theme----------
 
+
+
+API_TOKEN = os.getenv("HF_READ_TOKEN")
+headers = {"Authorization": f"Bearer {API_TOKEN}"}
+timeout = 100
+
 def flip_image(x):
     return np.fliplr(x)
 
@@ -284,8 +292,31 @@ css = """
 with gr.Blocks(theme=theme, css=css) as app:
     gr.HTML("<center><h6>🎨 Image Studio</h6></center>")
     with gr.Tab("Text to Image"):
-     # gr.load("models/digiplay/AnalogMadness-realistic-model-v7")
-        gr.load("models/XLabs-AI/flux-RealismLora")
+        #gr.load("models/digiplay/AnalogMadness-realistic-model-v7")
+        #gr.load("models/XLabs-AI/flux-RealismLora")
+    endpoint = HfInferenceEndpoint(
+        repo_id="XLabs-AI/flux-RealismLora", 
+        task="text-to-image", 
+        #token=API_KEY
+    )
+
+    def generate_image(text):
+        response = endpoint.text_to_image(
+            text,
+            guidance_scale=5,  # Adjust as needed
+            steps=50,          # Adjust as needed
+            height=1152,        # Adjust as needed
+            width=896          # Adjust as needed
+        )
+        return response
+
+    iface = gr.Interface(
+        fn=generate_image,
+        inputs=gr.Textbox(lines=2, placeholder="Enter your prompt here..."),
+        outputs=gr.Image(),
+        title="Flux Text-to-Image",
+        description="Generate images using the XLabs-AI/flux-RealismLora model.",
+    )
     
     with gr.Tab("Flip Image"):
                 with gr.Row():