Rioo26 commited on
Commit
7509ca1
·
verified ·
1 Parent(s): 821b5fa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,12 +9,12 @@ model = AutoModelForCausalLM.from_pretrained("microsoft/biogpt")
9
  def chat(prompt):
10
  inputs = tokenizer(prompt, return_tensors="pt")
11
  with torch.no_grad():
12
- outputs = model.generate(**inputs, max_new_tokens=100, do_sample=True)
13
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
14
 
15
  gr.Interface(
16
  fn=chat,
17
  inputs=gr.Textbox(label="输入医学问题或症状描述"),
18
  outputs=gr.Textbox(),
19
- title="医疗语言模型"
20
  ).launch()
 
9
  def chat(prompt):
10
  inputs = tokenizer(prompt, return_tensors="pt")
11
  with torch.no_grad():
12
+ outputs = model.generate(**inputs, do_sample=True)
13
  return tokenizer.decode(outputs[0], skip_special_tokens=True)
14
 
15
  gr.Interface(
16
  fn=chat,
17
  inputs=gr.Textbox(label="输入医学问题或症状描述"),
18
  outputs=gr.Textbox(),
19
+ title="医疗语言模型BioGPT"
20
  ).launch()