Spaces:
Runtime error
Runtime error
Boning c
commited on
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,8 +6,8 @@ import time
|
|
| 6 |
from html import escape
|
| 7 |
|
| 8 |
# Model config
|
| 9 |
-
PRIMARY_MODEL = "Smilyai-labs/Sam-reason-
|
| 10 |
-
FALLBACK_MODEL = "Smilyai-labs/Sam-reason-
|
| 11 |
USAGE_LIMIT = 5
|
| 12 |
RESET_AFTER_SECONDS = 20 * 60 # 20 minutes
|
| 13 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
@@ -85,16 +85,16 @@ def respond(message, history, reasoning_enabled, request: gr.Request):
|
|
| 85 |
for output in generate_stream(prompt, use_fallback=use_fallback):
|
| 86 |
formatted = format_thinking(output)
|
| 87 |
history[-1][1] = f"{formatted}<br><sub style='color:gray'>({model_used})</sub>"
|
| 88 |
-
yield history, history, f"π§
|
| 89 |
|
| 90 |
def clear_chat():
|
| 91 |
-
return [], [], "π§
|
| 92 |
|
| 93 |
# UI
|
| 94 |
with gr.Blocks() as demo:
|
| 95 |
gr.Markdown("# π€ SamAI β Reasoning Chat")
|
| 96 |
model_status = gr.Textbox(interactive=False, label="Model Status")
|
| 97 |
-
usage_counter = gr.Textbox(value="π§
|
| 98 |
chat_box = gr.Chatbot(type="tuples")
|
| 99 |
chat_state = gr.State([])
|
| 100 |
|
|
|
|
| 6 |
from html import escape
|
| 7 |
|
| 8 |
# Model config
|
| 9 |
+
PRIMARY_MODEL = "Smilyai-labs/Sam-reason-A3"
|
| 10 |
+
FALLBACK_MODEL = "Smilyai-labs/Sam-reason-A1"
|
| 11 |
USAGE_LIMIT = 5
|
| 12 |
RESET_AFTER_SECONDS = 20 * 60 # 20 minutes
|
| 13 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
|
| 85 |
for output in generate_stream(prompt, use_fallback=use_fallback):
|
| 86 |
formatted = format_thinking(output)
|
| 87 |
history[-1][1] = f"{formatted}<br><sub style='color:gray'>({model_used})</sub>"
|
| 88 |
+
yield history, history, f"π§ A3 messages left: {remaining}"
|
| 89 |
|
| 90 |
def clear_chat():
|
| 91 |
+
return [], [], "π§ A3 messages left: 5"
|
| 92 |
|
| 93 |
# UI
|
| 94 |
with gr.Blocks() as demo:
|
| 95 |
gr.Markdown("# π€ SamAI β Reasoning Chat")
|
| 96 |
model_status = gr.Textbox(interactive=False, label="Model Status")
|
| 97 |
+
usage_counter = gr.Textbox(value="π§ A3 messages left: 10", interactive=False, show_label=False)
|
| 98 |
chat_box = gr.Chatbot(type="tuples")
|
| 99 |
chat_state = gr.State([])
|
| 100 |
|