rieon commited on
Commit
5780d63
·
1 Parent(s): f96d067
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -35,15 +35,17 @@ def respond(
35
  yield generated
36
 
37
  demo = gr.ChatInterface(
38
- fn=respond,
39
- system_message="You are a helpful coding assistant.",
 
 
 
40
  additional_inputs=[
41
  gr.Textbox(value="You are a helpful coding assistant.", label="System message"),
42
  gr.Slider(1, 2048, value=512, step=1, label="Max new tokens"),
43
  gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Temperature"),
44
  gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p"),
45
  ],
46
- title="DeepCoder-14B (LoRA Fine-Tuned)",
47
  )
48
 
49
  if __name__ == "__main__":
 
35
  yield generated
36
 
37
  demo = gr.ChatInterface(
38
+ respond,
39
+ title="DeepCoder with Suger",
40
+ description="Upload any text or pdf files and ask questions about them!",
41
+ textbox=gr.MultimodalTextbox(file_types=[".pdf", ".txt"]),
42
+ multimodal=True,
43
  additional_inputs=[
44
  gr.Textbox(value="You are a helpful coding assistant.", label="System message"),
45
  gr.Slider(1, 2048, value=512, step=1, label="Max new tokens"),
46
  gr.Slider(0.1, 4.0, value=0.7, step=0.1, label="Temperature"),
47
  gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p"),
48
  ],
 
49
  )
50
 
51
  if __name__ == "__main__":