Meteord commited on
Commit
ec247b7
·
verified ·
1 Parent(s): 5af207c

Sync from GitHub via hub-sync

Browse files
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 = 512,
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,