vilarin commited on
Commit
2d251eb
1 Parent(s): bfe9f15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,7 +22,7 @@ MODEL_NAME = MODEL_ID.split("/")[-1]
22
 
23
  TITLE = "<h1><center>VL-Chatbox</center></h1>"
24
 
25
- DESCRIPTION = "<h3><center>MODEL: " + f'[{MODEL_NAME}](https://hf.co/models/{MODEL_NAME})' + "</center></h3>"
26
 
27
  CSS = """
28
  .duplicate-button {
@@ -69,7 +69,7 @@ def stream_chat(message, history: list, temperature: float, max_new_tokens: int)
69
 
70
  generate_kwargs = dict(
71
  image=image,
72
- msg=conversation,
73
  streamer=streamer,
74
  max_new_tokens=max_new_tokens,
75
  temperature=temperature,
 
22
 
23
  TITLE = "<h1><center>VL-Chatbox</center></h1>"
24
 
25
+ DESCRIPTION = f'<h3><center>MODEL: <a href="https://hf.co/models/{MODEL_ID}">{MODEL_NAME}</a></center></h3>'
26
 
27
  CSS = """
28
  .duplicate-button {
 
69
 
70
  generate_kwargs = dict(
71
  image=image,
72
+ msgs=conversation,
73
  streamer=streamer,
74
  max_new_tokens=max_new_tokens,
75
  temperature=temperature,