Spaces:
Runtime error
Runtime error
Commit ·
5aa708b
1
Parent(s): 677a99e
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,10 @@ def generate_script(host_name, listener_location, causes_climate_change, co2_lev
|
|
| 27 |
|
| 28 |
# Generate the script using the language model
|
| 29 |
script = generator.text_generation(prompt_template)[0]['generated_text']
|
| 30 |
-
|
|
|
|
|
|
|
|
|
|
| 31 |
# Split the script into sections
|
| 32 |
sections = script.split("\n")
|
| 33 |
|
|
|
|
| 27 |
|
| 28 |
# Generate the script using the language model
|
| 29 |
script = generator.text_generation(prompt_template)[0]['generated_text']
|
| 30 |
+
if isinstance(response, list):
|
| 31 |
+
script = response[0].get('generated_text', '')
|
| 32 |
+
else:
|
| 33 |
+
script = response.get('generated_text', '')
|
| 34 |
# Split the script into sections
|
| 35 |
sections = script.split("\n")
|
| 36 |
|