Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import cv2
 import numpy as np
 from datetime import datetime
+from diffusers import DiffusionPipeline
 import random
 
 #----------Start of theme----------
@@ -326,6 +327,10 @@ with gr.Blocks(theme=theme, css=css) as app:
     with gr.Tab("Text to Image"):
  
        # gr.load("models/digiplay/AnalogMadness-realistic-model-v7")
-        gr.load("models/XLabs-AI/flux-RealismLora")
+        pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev")
+        pipe.load_lora_weights("XLabs-AI/flux-RealismLora")
+
+        prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
+        image = pipe(prompt).images[0]
     
 app.launch(share=True)