Update app.py

app.py

@@ -24,19 +24,25 @@ def load_css():
         </style>""",
         unsafe_allow_html=True,
     )
-@st.cache_resource(show_spinner=False)
+    
+@st.cache_resource
 def load_all_models():
     from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+    from diffusers import StableDiffusionPipeline
+    import torch
 
+    # Translation model
     model_id = "ai4bharat/indictrans2-indic-en-dist-200M"
-
     tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
     model = AutoModelForSeq2SeqLM.from_pretrained(model_id, trust_remote_code=True)
 
+    # Text generation model (simple GPT-2)
     text_gen = pipeline("text-generation", model="gpt2")
+
+    # Image generation model (Stable Diffusion)
     img_pipe = StableDiffusionPipeline.from_pretrained(
-        "CompVis/stable-diffusion-v1-4", 
-        revision="fp16", 
+        "CompVis/stable-diffusion-v1-4",
+        revision="fp16",
         torch_dtype=torch.float16
     ).to("cuda")