Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
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
|