Update app.py

app.py

@@ -13,9 +13,9 @@ except ImportError:
     from llama_cpp import Llama
 
 # Initialize model
-model_path = "AstroSage-8B-BF16.gguf"  # Downloaded from AstroMLab/AstroSage-8B-GGUF
+model_path = "tinyllama-1.1b-chat-v1.0.Q4_0.gguf"  # Downloaded from TinyLlama/TinyLlama-1.1B-Chat-v1.0-GGUF
 llm = Llama.from_pretrained(
-    repo_id="AstroMLab/AstroSage-8B-GGUF",
+    repo_id="TinyLlama/TinyLlama-1.1B-Chat-v1.0-GGUF",
     filename=model_path,
     n_ctx=2048,  # Context length for prompts
     n_threads=2  # Use 2 CPU cores
@@ -23,14 +23,14 @@ llm = Llama.from_pretrained(
 
 def generate_astrology_prediction(prompt: str) -> Generator[str, None, None]:
     """
-    Generates astrology-based fortune-telling predictions using AstroSage-8B-BF16.gguf with streaming.
+    Generates astrology-based fortune-telling predictions using TinyLlama-1.1B-Chat-v1.0 with streaming.
     """
     system_prompt = (
         "You are an expert astrologer, specializing in fortune-telling. Given a user prompt "
         "containing details like zodiac sign, birth date, or specific questions, provide predictions "
         "about their future, career, love life, and success. Stream the output line by line. "
         "Use bullet points for key predictions and keep responses engaging and concise. "
-        "Despite being trained on astronomy, adapt your knowledge to provide astrology-like insights."
+        "If the prompt is vague (e.g., 'Hi'), ask for more details like zodiac sign or birth date."
     )
     full_prompt = f"<|SYSTEM|> {system_prompt}\n<|USER|> {prompt}\n<|ASSISTANT|>"