Spaces:
Running
Running
Deepak Yadav
commited on
Commit
·
7cf558f
1
Parent(s):
7d988b6
updated new version deepseek-r1
Browse files- services/llm.py +1 -1
services/llm.py
CHANGED
@@ -8,7 +8,7 @@ def initialize_llm(model_name, temperature, top_p, max_tokens):
|
|
8 |
# Configure the LLM with additional parameters
|
9 |
llm = OllamaLLM(
|
10 |
model=model_name,
|
11 |
-
|
12 |
temperature=temperature, # Controls randomness (0 = deterministic, 1 = max randomness)
|
13 |
max_tokens=max_tokens, # Limit the number of tokens in the output
|
14 |
top_p=top_p # Nucleus sampling for controlling diversity
|
|
|
8 |
# Configure the LLM with additional parameters
|
9 |
llm = OllamaLLM(
|
10 |
model=model_name,
|
11 |
+
base_url="https://deepak7376-ollama-server.hf.space",
|
12 |
temperature=temperature, # Controls randomness (0 = deterministic, 1 = max randomness)
|
13 |
max_tokens=max_tokens, # Limit the number of tokens in the output
|
14 |
top_p=top_p # Nucleus sampling for controlling diversity
|