Spaces:
Runtime error
Runtime error
Commit
•
6d22485
1
Parent(s):
8667c9e
Update ollama_serve.py
Browse files- ollama_serve.py +6 -2
ollama_serve.py
CHANGED
|
@@ -1,8 +1,12 @@
|
|
| 1 |
import os
|
|
|
|
| 2 |
|
| 3 |
-
# model name
|
| 4 |
-
model =
|
|
|
|
|
|
|
| 5 |
os.system('ollama serve &')
|
|
|
|
| 6 |
os.system(f'ollama pull {model}')
|
| 7 |
# verify the contents
|
| 8 |
os.system(f'ollama pull {model}')
|
|
|
|
| 1 |
import os
|
| 2 |
+
import sys
|
| 3 |
|
| 4 |
+
# get model name
|
| 5 |
+
model = sys.argv[1]
|
| 6 |
+
|
| 7 |
+
# start the server
|
| 8 |
os.system('ollama serve &')
|
| 9 |
+
# pull the model
|
| 10 |
os.system(f'ollama pull {model}')
|
| 11 |
# verify the contents
|
| 12 |
os.system(f'ollama pull {model}')
|