Update app.py

app.py

@@ -3,13 +3,14 @@
 from transformers import pipeline
 import gradio as gr
 from diffusers import DiffusionPipeline
+from diffusers import FluxPipeline
 import torch
 import time
 
 # Load models
 translator = pipeline("translation", model="Helsinki-NLP/opus-mt-dra-en")
 summarizer = pipeline("summarization", model="Falconsai/text_summarization")
-image_pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float16).to("cpu")
+image_pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.float16).to("cpu")
 
 
 # Translate Tamil to English