nagarmayank commited on
Commit
333660b
·
1 Parent(s): f45803c

1b changed

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -126,7 +126,7 @@ def write_message(data: dict):
126
 
127
  @app.get("/ask")
128
  def ask(prompt: str):
129
- model = ChatOllama(model="gemma3:4b", temperature=1)
130
  return model.invoke(prompt)
131
 
132
  if __name__ == "__main__":
 
126
 
127
  @app.get("/ask")
128
  def ask(prompt: str):
129
+ model = ChatOllama(model="gemma3:1b", temperature=1)
130
  return model.invoke(prompt)
131
 
132
  if __name__ == "__main__":