tien314 commited on
Commit
36f0086
·
verified ·
1 Parent(s): b59b2ba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +36,7 @@ def load_model():
36
  #llm = OllamaLLM(model="gemma2", temperature=0, device=device)
37
  #api_key = "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"
38
  api_key = "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"
39
- llm = ChatGroq(model = "llama-3.3-70b-versatile", temperature = 0,api_key = api_key)
40
  chain = prompt|llm
41
  return chain
42
 
 
36
  #llm = OllamaLLM(model="gemma2", temperature=0, device=device)
37
  #api_key = "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"
38
  api_key = "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"
39
+ llm = ChatGroq(model = "gemma2-9b-it", temperature = 0,api_key = api_key)
40
  chain = prompt|llm
41
  return chain
42