Update app.py

app.py

@@ -4,16 +4,13 @@ import os
 from dotenv import load_dotenv
 import warnings
 
-# Suppress CUDA warnings
-warnings.filterwarnings('ignore', category=UserWarning, module='torch.cuda')
+# Suppress warnings
+warnings.filterwarnings('ignore', category=UserWarning)
 
 # Load environment variables
 load_dotenv()
 HF_TOKEN = os.getenv("HF_TOKEN")
 
-# Check CUDA availability and set device
-DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
-
 def generate_diagram(prompt, width=1024, height=1024):
     """Generate a diagram using FLUX AI"""
     try:
@@ -35,6 +32,7 @@ def generate_diagram(prompt, width=1024, height=1024):
         raise gr.Error(f"Error generating diagram: {str(e)}")
 
 
+
 # Enhanced examples with more detailed prompts and specific styling
 EXAMPLES = [
     {
@@ -238,8 +236,6 @@ GRADIO_EXAMPLES = [
     for example in EXAMPLES
 ]
 
-# Create Gradio interface
-# Gradio interface 부분 수정
 demo = gr.Interface(
     fn=generate_diagram,
     inputs=[
@@ -283,16 +279,15 @@ demo = gr.Interface(
         └── DETAIL 2 [Icon]
     ```
     """,
-    examples=GRADIO_EXAMPLES,  # EXAMPLES 대신 GRADIO_EXAMPLES 사용
+    examples=GRADIO_EXAMPLES,
     cache_examples=True,
-    theme=gr.themes.Soft(),
-    allow_flagging="never"  # 불필요한 플래깅 기능 제거
+    theme=gr.themes.Soft()
 )
 
-# Launch 부분만 수정
+# Launch app
 if __name__ == "__main__":
     try:
-        demo.queue(max_size=10)  # 큐 설정은 launch 전에
+        demo.queue(max_size=10)
         demo.launch(
             server_name="0.0.0.0",
             server_port=7860,