RAHMAN00700 commited on
Commit
72ba5bf
1 Parent(s): afa266c

Update run.py

Browse files
Files changed (1) hide show
  1. run.py +1 -1
run.py CHANGED
@@ -9,7 +9,7 @@ from flask import Flask
9
  import ollama
10
 
11
  # Optionally specify the model to pull during startup:
12
- model_name = "llama3" # Replace with the desired model name
13
  import os
14
  import subprocess
15
  # List of allowed models
 
9
  import ollama
10
 
11
  # Optionally specify the model to pull during startup:
12
+ model_name = "gemma:2b" # Replace with the desired model name
13
  import os
14
  import subprocess
15
  # List of allowed models