Update app.py

app.py

@@ -4,16 +4,15 @@ import os
 from typing import Generator
 
 # Initialize Inference API client
-# Try AstroSage-Llama-3.1-8B; fallback to Llama-3.2-3B-Instruct if unsupported
-model = "AstroMLab/AstroSage-Llama-3.1-8B"  # Replace with "meta-llama/Llama-3.2-3B-Instruct" if needed
+model = "xAI/grok-3.2-mini"  # Conversational model for astronomy queries
 client = InferenceClient(model=model, token=os.getenv("HF_TOKEN"))
 
 def generate_astronomy_explanation(prompt: str) -> Generator[str, None, None]:
     """
-    Generates astronomy-related explanations using the model via Inference API with streaming.
+    Generates astronomy-related explanations using xAI/grok-3.2-mini via Inference API with streaming.
     """
     system_prompt = (
-        "You are AstroSage, an expert astronomy assistant like Grok. Given a user prompt about "
+        "You are an expert astronomy assistant like Grok. Given a user prompt about "
         "astronomical phenomena, space exploration, stars, galaxies, cosmology, planetary science, "
         "or astrophysics, provide a clear, educational explanation. Stream the output line by line. "
         "Use bullet points for key points and keep responses concise yet informative."