Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,13 +29,14 @@ runner = STORMWikiRunner(engine_args, lm_configs, rm)
|
|
| 29 |
|
| 30 |
@spaces.GPU
|
| 31 |
def generate_article(prompt, progress=gr.Progress(track_tqdm=True)):
|
| 32 |
-
runner.run(
|
| 33 |
topic=prompt,
|
| 34 |
do_research=True,
|
| 35 |
do_generate_outline=True,
|
| 36 |
do_generate_article=True,
|
| 37 |
do_polish_article=True,
|
| 38 |
)
|
|
|
|
| 39 |
runner.post_run()
|
| 40 |
runner.summary()
|
| 41 |
|
|
|
|
| 29 |
|
| 30 |
@spaces.GPU
|
| 31 |
def generate_article(prompt, progress=gr.Progress(track_tqdm=True)):
|
| 32 |
+
response = runner.run(
|
| 33 |
topic=prompt,
|
| 34 |
do_research=True,
|
| 35 |
do_generate_outline=True,
|
| 36 |
do_generate_article=True,
|
| 37 |
do_polish_article=True,
|
| 38 |
)
|
| 39 |
+
print(response)
|
| 40 |
runner.post_run()
|
| 41 |
runner.summary()
|
| 42 |
|