Update app.py
Browse files
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 = "
|
| 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 |
|