Update app.py

app.py

@@ -6,6 +6,7 @@ import requests
 import tiktoken
 
 # from ollama import ollama as Client
+import os 
 from functools import wraps
 from collections import defaultdict
 from flask import Flask, request, jsonify, Response
@@ -27,7 +28,7 @@ def run_command(command):
         text=True
     )
     for line in process.stdout:
-        yield line
+        print(line)
     process.wait()
     if process.returncode != 0:
         sys.exit(process.returncode)
@@ -476,7 +477,6 @@ if __name__ == "__main__":
     "gemma3:4b"
 ]
     for model in models:
-        for chunk in client.pull_model(models):
-            print(chunk)
-            
+        run_command(f"ollama pull {model}")
+
     app.run(host=HOST, port=7860)