Update app.py

app.py

@@ -3,18 +3,16 @@ from transformers import pipeline
 from diffusers import StableDiffusionPipeline
 import torch
 
-# Check for GPU availability
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-# Set appropriate dtype based on device
 torch_dtype = torch.float16 if device == "cuda" else torch.float32
 
-# Load sentiment analysis model
+# Sentiment Analysis Model
 @st.cache_resource
 def load_sentiment_analyzer():
     return pipeline("text-classification", model="SamLowe/roberta-base-go_emotions", top_k=5, device=device)
 
-# Load text-to-image model
+# Text-to-image model
 @st.cache_resource
 def load_text2img():
     model = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch_dtype, safety_checker = None, requires_safety_checker = False)