Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,8 @@ model = Llama(
|
|
| 8 |
model_path=hf_hub_download(
|
| 9 |
repo_id=os.environ.get("REPO_ID", "Lyte/QuadConnect2.5-0.5B-v0.0.4b"),
|
| 10 |
filename=os.environ.get("MODEL_FILE", "unsloth.Q8_0.gguf"),
|
| 11 |
-
)
|
|
|
|
| 12 |
)
|
| 13 |
|
| 14 |
SYSTEM_PROMPT = """You are an expert Connect Four player. The game is played on a 6x7 grid where pieces fall to the lowest available position in each column due to gravity.
|
|
@@ -237,7 +238,7 @@ def create_interface():
|
|
| 237 |
],
|
| 238 |
temperature=0.8,
|
| 239 |
top_p=0.95,
|
| 240 |
-
|
| 241 |
)
|
| 242 |
|
| 243 |
ai_response = response['choices'][0]['message']['content']
|
|
|
|
| 8 |
model_path=hf_hub_download(
|
| 9 |
repo_id=os.environ.get("REPO_ID", "Lyte/QuadConnect2.5-0.5B-v0.0.4b"),
|
| 10 |
filename=os.environ.get("MODEL_FILE", "unsloth.Q8_0.gguf"),
|
| 11 |
+
),
|
| 12 |
+
n_ctx=16384
|
| 13 |
)
|
| 14 |
|
| 15 |
SYSTEM_PROMPT = """You are an expert Connect Four player. The game is played on a 6x7 grid where pieces fall to the lowest available position in each column due to gravity.
|
|
|
|
| 238 |
],
|
| 239 |
temperature=0.8,
|
| 240 |
top_p=0.95,
|
| 241 |
+
max_tokens=2048
|
| 242 |
)
|
| 243 |
|
| 244 |
ai_response = response['choices'][0]['message']['content']
|