Update app.py

app.py

@@ -4,6 +4,7 @@ import numpy as np
 import torch
 from datetime import datetime
 from diffusers import DiffusionPipeline
+from transformers import pipeline
 import random
 
 #----------Start of theme----------
@@ -287,9 +288,23 @@ 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")
-        image_input = gr.Image(type="numpy", label="Upload Image")       
-        image_output = gr.Image(format="png")        
+        #gr.load("models/XLabs-AI/flux-RealismLora")
+# Load the text-to-image model from Hugging Face
+        generator = pipeline("text-to-image", model="XLabs-AI/flux-RealismLora")
+
+# Define the generation function
+    def generate_image(prompt):
+        result = generator(prompt)
+        return result[0]['generated_image']
+
+# Set up the Gradio Interface
+demo = gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(label="Enter a prompt for the image"),
+    outputs=gr.Image(label="Generated Image"),
+    title="Text-to-Image Generation with Flux-RealismLora",
+    description="Enter a text prompt to generate a realistic image using the XLabs-AI/flux-RealismLora model."
+)        
     with gr.Tab("Flip Image"):
                 with gr.Row():
                     image_input = gr.Image(type="numpy", label="Upload Image")