Sync from GitHub via hub-sync
Browse files- app/app.py +9 -0
- app/backend/smolnalysis_model_wrapper.py +1 -1
app/app.py
CHANGED
|
@@ -50,6 +50,14 @@ DEFAULT_TEMPERATURE = float(os.getenv("SMOLNALYSIS_MINICPM_TEMPERATURE", "0.7"))
|
|
| 50 |
DEFAULT_TOP_P = float(os.getenv("SMOLNALYSIS_MINICPM_TOP_P", "0.95"))
|
| 51 |
DEFAULT_TOP_K = int(os.getenv("SMOLNALYSIS_MINICPM_TOP_K", "64"))
|
| 52 |
LOAD_IN_4BIT = os.getenv("SMOLNALYSIS_MINICPM_LOAD_IN_4BIT", "true").casefold() not in {"0", "false", "no", "off"}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 53 |
|
| 54 |
OPENUI_PREVIEW_CSS = """
|
| 55 |
.preview { border: 1px solid #d7dde8; border-radius: 8px; padding: 14px; background: #fff; color: #111827; }
|
|
@@ -231,6 +239,7 @@ with gr.Blocks(title="smolnalysis") as app:
|
|
| 231 |
)
|
| 232 |
send = gr.Button("Send", variant="primary", scale=1)
|
| 233 |
clear = gr.Button("Clear", scale=1)
|
|
|
|
| 234 |
|
| 235 |
prompt.submit(submit_message, [prompt, chatbot, model_messages], [prompt, chatbot, model_messages])
|
| 236 |
send.click(submit_message, [prompt, chatbot, model_messages], [prompt, chatbot, model_messages])
|
|
|
|
| 50 |
DEFAULT_TOP_P = float(os.getenv("SMOLNALYSIS_MINICPM_TOP_P", "0.95"))
|
| 51 |
DEFAULT_TOP_K = int(os.getenv("SMOLNALYSIS_MINICPM_TOP_K", "64"))
|
| 52 |
LOAD_IN_4BIT = os.getenv("SMOLNALYSIS_MINICPM_LOAD_IN_4BIT", "true").casefold() not in {"0", "false", "no", "off"}
|
| 53 |
+
EXAMPLE_PROMPTS = [
|
| 54 |
+
["Erstelle ein Monatsdiagramm für Bearbeitungszeit in Isar."],
|
| 55 |
+
["Zeige mir Heizbedarf pro Monat für 2023 in Bürgerbüro Pasing."],
|
| 56 |
+
["Welche Werte gibt es für Einwohner in den Bezirken?"],
|
| 57 |
+
["Prüfe den Grenzwert für Stromverbrauch in Bogenhausen."],
|
| 58 |
+
["Liste die Werte nach Dienststelle auf."],
|
| 59 |
+
["Summarize a Python function for a teammate."],
|
| 60 |
+
]
|
| 61 |
|
| 62 |
OPENUI_PREVIEW_CSS = """
|
| 63 |
.preview { border: 1px solid #d7dde8; border-radius: 8px; padding: 14px; background: #fff; color: #111827; }
|
|
|
|
| 239 |
)
|
| 240 |
send = gr.Button("Send", variant="primary", scale=1)
|
| 241 |
clear = gr.Button("Clear", scale=1)
|
| 242 |
+
gr.Examples(examples=EXAMPLE_PROMPTS, inputs=prompt)
|
| 243 |
|
| 244 |
prompt.submit(submit_message, [prompt, chatbot, model_messages], [prompt, chatbot, model_messages])
|
| 245 |
send.click(submit_message, [prompt, chatbot, model_messages], [prompt, chatbot, model_messages])
|
app/backend/smolnalysis_model_wrapper.py
CHANGED
|
@@ -276,7 +276,7 @@ class SmolnalysisMoE(torch.nn.Module):
|
|
| 276 |
self,
|
| 277 |
inputs: Any,
|
| 278 |
*,
|
| 279 |
-
max_new_tokens: int =
|
| 280 |
temperature: float = 0.7,
|
| 281 |
top_p: float = 0.95,
|
| 282 |
top_k: int = 64,
|
|
|
|
| 276 |
self,
|
| 277 |
inputs: Any,
|
| 278 |
*,
|
| 279 |
+
max_new_tokens: int = 1024,
|
| 280 |
temperature: float = 0.7,
|
| 281 |
top_p: float = 0.95,
|
| 282 |
top_k: int = 64,
|