Update app.py

app.py

@@ -1,10 +1,7 @@
 import gradio as gr
 import cv2
 import numpy as np
-import torch
 from datetime import datetime
-from diffusers import DiffusionPipeline
-from transformers import pipeline
 import random
 
 #----------Start of theme----------
@@ -287,31 +284,17 @@ 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")
-# 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."
-)        
+     # gr.load("models/digiplay/AnalogMadness-realistic-model-v7")
+    gr.load("models/XLabs-AI/flux-RealismLora")
+    
     with gr.Tab("Flip Image"):
-        with gr.Row():
-            image_input = gr.Image(type="numpy", label="Upload Image")
-            image_output = gr.Image(format="png")
-        with gr.Row():    
-             image_button = gr.Button("Run", variant='primary')
-             image_button.click(flip_image, inputs=image_input, outputs=image_output)
+                with gr.Row():
+                    image_input = gr.Image(type="numpy", label="Upload Image")
+                    image_output = gr.Image(format="png")
+                with gr.Row():    
+                    image_button = gr.Button("Run", variant='primary')
+                    image_button.click(flip_image, inputs=image_input, outputs=image_output)
+                    
     with gr.Tab("Image Filters"):    
         with gr.Row():
             with gr.Column():
@@ -347,4 +330,4 @@ demo = gr.Interface(
             )
 
     
-app.launch(share=False)
+app.launch(share=True)