FawadHaider2 commited on
Commit
5d30eea
·
verified ·
1 Parent(s): 168d3b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -85,7 +85,7 @@ def generate_information(disease_info, model_name):
85
  pipe = load_model(model_name)
86
 
87
  # Generate information
88
- result = pipe(disease_info, max_length=50, num_return_sequences=1)
89
  return result[0]['generated_text']
90
 
91
  # Create Streamlit Interface with Sidebar
 
85
  pipe = load_model(model_name)
86
 
87
  # Generate information
88
+ result = pipe(disease_info, max_length=200, num_return_sequences=1)
89
  return result[0]['generated_text']
90
 
91
  # Create Streamlit Interface with Sidebar