Praisepaul commited on
Commit
37342ae
·
verified ·
1 Parent(s): 4cde872

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ bio_gpt = pipeline("text-generation", model="microsoft/biogpt")
5
 
6
  def medical_chatbot(query):
7
  try:
8
- result = bio_gpt(query, max_length=200, num_return_sequences=1, temperature=0.7, top_k=50)
9
  generated_text = result[0]["generated_text"][len(query):].strip()
10
  return generated_text
11
  except Exception as e:
 
5
 
6
  def medical_chatbot(query):
7
  try:
8
+ result = bio_gpt(query, max_length=1000, num_return_sequences=1, temperature=0.7, top_k=50)
9
  generated_text = result[0]["generated_text"][len(query):].strip()
10
  return generated_text
11
  except Exception as e: