MaxLSB commited on
Commit
90d1b16
·
verified ·
1 Parent(s): 0472769

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -24,7 +24,7 @@ load_model("LeCarnet-8M")
24
  # Streaming generation function
25
  def respond(message, max_tokens, temperature, top_p):
26
  inputs = tokenizer(message, return_tensors="pt")
27
- streamer = TextIteratorStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
28
 
29
  generate_kwargs = dict(
30
  **inputs,
@@ -66,7 +66,7 @@ def update_model(model_name):
66
  with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
67
  with gr.Row(equal_height=True):
68
  gr.Markdown(
69
- f'<img src="file/{os.path.abspath("media/le-carnet.png")}" style="height:80px; width:80px; margin-right:10px; vertical-align:top;">',
70
  elem_classes="header-image"
71
  )
72
  # Title next to image
 
24
  # Streaming generation function
25
  def respond(message, max_tokens, temperature, top_p):
26
  inputs = tokenizer(message, return_tensors="pt")
27
+ streamer = TextIteratorStreamer(tokenizer, skip_prompt=False, skip_special_tokens=True)
28
 
29
  generate_kwargs = dict(
30
  **inputs,
 
66
  with gr.Blocks(title="LeCarnet - Chat Interface") as demo:
67
  with gr.Row(equal_height=True):
68
  gr.Markdown(
69
+ f'<img src="media/le-carnet.png" style="height:80px; width:80px; margin-right:10px; vertical-align:top;">',
70
  elem_classes="header-image"
71
  )
72
  # Title next to image