Update app.py

app.py

@@ -19,13 +19,13 @@ MODEL_NAME = "deepseek-llm-7b"
 MODEL_PATH = f"models/{MODEL_NAME}"
 
 def download_model():
-    if not os.path.exists(MODEL_PATH):
-        print(f"🚀 Downloading model: {MODEL_NAME} to {MODEL_PATH} ...")
-        os.makedirs("models", exist_ok=True)
-        subprocess.run(["ollama", "pull", f"deepseek/{MODEL_NAME}"], check=True)
-        print(f"✅ Model downloaded to {MODEL_PATH}.")
+    models = ollama.list()
+    if any(m["name"] == MODEL_NAME for m in models["models"]):
+        print(f"✅ Model '{MODEL_NAME}' is already available.")
     else:
-        print(f"✅ Model {MODEL_NAME} already exists.")
+        print(f"🚀 Downloading model: {MODEL_NAME} ...")
+        ollama.pull(MODEL_NAME)
+        print(f"✅ Model '{MODEL_NAME}' downloaded successfully.")
 
 # Generate AI response using Ollama
 def chat_response(user_input):