SkyNetWalker commited on
Commit
838508d
·
verified ·
1 Parent(s): c8442ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import json
5
  # Define the URL for the local Ollama API and the model name
6
  OLLAMA_API_URL = "http://localhost:11434/api/generate"
7
  # This must match the name used in `ollama pull` in Dockerfile
8
- MODEL_NAME = "hf.co/unsloth/gemma-3-4b-it-qat-GGUF:Q4_K_M"
9
 
10
  def generate_text(prompt, max_new_tokens=256, temperature=0.7):
11
  """
 
5
  # Define the URL for the local Ollama API and the model name
6
  OLLAMA_API_URL = "http://localhost:11434/api/generate"
7
  # This must match the name used in `ollama pull` in Dockerfile
8
+ MODEL_NAME = "gemma3_4b_it_qat"
9
 
10
  def generate_text(prompt, max_new_tokens=256, temperature=0.7):
11
  """