Update app.py

app.py

@@ -1,12 +1,10 @@
-# hugging face
 import gradio as gr
 import requests
 from transformers import MarianMTModel, MarianTokenizer, AutoModelForCausalLM, AutoTokenizer
 from PIL import Image
 import torch
-import os
 import io
-
+import os
 # Check if GPU is available
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
@@ -23,7 +21,7 @@ if generator_tokenizer.pad_token is None:
     generator_tokenizer.pad_token = generator_tokenizer.eos_token
 
 # Hugging Face API for Image Generation
-HF_API_KEY = os.getenv("HF_API_KEY")  # Use environment variable
+HF_API_KEY = os.getenv("HF_API_KEY")
 IMAGE_GEN_URL = "https://api-inference.huggingface.co/models/black-forest-labs/FLUX.1-schnell"
 HEADERS = {"Authorization": f"Bearer {HF_API_KEY}"}
 
@@ -67,4 +65,4 @@ interface = gr.Interface(
 )
 
 # Launch the Gradio app
-interface.launch()
+interface.launch()