Update app.py

app.py

@@ -1,12 +1,8 @@
 import gradio as gr
 import cv2
 import numpy as np
-import os
 from datetime import datetime
 import random
-from transformers import pipeline
-import torch
-from diffusers import FluxPipeline
 
 #----------Start of theme----------
 theme = gr.themes.Soft(
@@ -42,12 +38,6 @@ 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)
 
@@ -280,6 +270,7 @@ def image_processing(image, filter_type):
         output = image
         
     return cv2.cvtColor(output, cv2.COLOR_BGR2RGB) if len(output.shape) == 3 else output
+    
 
 css = """
 #app-container {
@@ -292,17 +283,20 @@ css = """
 # Gradio interface
 with gr.Blocks(theme=theme, css=css) as app:
     gr.HTML("<center><h6>🎨 Image Studio</h6></center>")
-    with gr.Tab("Text to Image"):
+        with gr.Tab("Text to Image"):
         #gr.HTML("<center><b>Flux</b></center>")
         #gr.load("models/XLabs-AI/flux-RealismLora")
         #gr.load("models/digiplay/AnalogMadness-realistic-model-v7")
         #gr.load("stabilityai/stable-diffusion-3-medium-diffusers")
-        model=gr.load("models/dvyio/flux-lora-film-noir")
-  
-with gr.Tab("Flip Image"):
-    with gr.Row():
-        image_input = gr.Image(type="numpy", label="Upload Image")
-        image_output = gr.Image(format="png")
+        gr.load("models/dvyio/flux-lora-film-noir")
+    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_button = gr.Button("Run", variant='primary')
         image_button.click(flip_image, inputs=image_input, outputs=image_output)