Update app.py

app.py

@@ -14,14 +14,15 @@ if not GROQ_API_KEY:
 # Initialize Groq client
 client = Groq(api_key=GROQ_API_KEY)
 
-# Set up device and load Stable Diffusion model
+# Set up device and load the exact model you specified
 device = "cuda" if torch.cuda.is_available() else "cpu"
+model_id = "sd-legacy/stable-diffusion-v1-5"  # Your exact model ID
 pipe = StableDiffusionPipeline.from_pretrained(
-    "sd-legacy/stable-diffusion-v1-5",
-    torch_dtype=torch.float16,
-    safety_checker=None,  # Optional: disable NSFW filter (comment out if unwanted)
+    model_id,
+    torch_dtype=torch.float16,  # Matches your snippet
+    safety_checker=None,  # Optional: disabled for flexibility
 ).to(device)
-pipe.enable_model_cpu_offload()  # Optimize memory usage
+pipe.enable_model_cpu_offload()  # Optimize memory for Spaces
 
 # Function to generate tutor output (lesson, question, feedback)
 def generate_tutor_output(subject, difficulty, student_input):
@@ -50,7 +51,7 @@ def generate_tutor_output(subject, difficulty, student_input):
     )
     return completion.choices[0].message.content
 
-# Function to generate images using sd-legacy/stable-diffusion-v1-5
+# Function to generate images using your model ID
 def generate_images(text, selected_model):
     if selected_model == "Stable Diffusion (Realism)":
         prompt_prefix = "realistic, detailed, vivid colors, "
@@ -62,7 +63,7 @@ def generate_images(text, selected_model):
     results = []
     for i in range(3):
         modified_text = f"{prompt_prefix}{text} variation {i+1}, high quality"
-        image = pipe(modified_text, num_inference_steps=25).images[0]
+        image = pipe(modified_text, num_inference_steps=25).images[0]  # Using your model
         results.append(image)
     return results
 
@@ -110,11 +111,11 @@ with gr.Blocks(title="AI Tutor with Visuals") as demo:
 
     gr.Markdown("""
     ### How to Use
-    1. **Text Section**: Select a subject and difficulty, type your query, and click 'Generate Lesson & Question' to get your personalized lesson, question, and feedback.
-    2. **Visual Section**: Select an image style and click 'Generate Visuals' to see 3 variations using Stable Diffusion v1.5 (free & open-source).
-    3. Review the AI-generated content to enhance your learning!
+    1. **Text Section**: Select a subject and difficulty, type your query, and click 'Generate Lesson & Question'.
+    2. **Visual Section**: Select a style and click 'Generate Visuals' to see 3 images from sd-legacy/stable-diffusion-v1-5.
+    3. Review the AI-generated content!
 
-    *Note*: Powered by sd-legacy/stable-diffusion-v1-5. GPU recommended for faster results.
+    *Example*: Try "a photo of an astronaut riding a horse on mars" for visuals.
     """)
 
     def process_output_text(subject, difficulty, student_input):