Spaces:
Sleeping
Sleeping
minor change in app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ description = """
|
|
| 12 |
|
| 13 |
prompt = 'Write a story. In the story, try to use the verb "fight", the noun "king" and the adjective "brave". Possible story:'
|
| 14 |
|
| 15 |
-
def generate_text(input_prompt=
|
| 16 |
print('inputs: ')
|
| 17 |
print('prompt:', prompt)
|
| 18 |
print('max_len:', max_len)
|
|
|
|
| 12 |
|
| 13 |
prompt = 'Write a story. In the story, try to use the verb "fight", the noun "king" and the adjective "brave". Possible story:'
|
| 14 |
|
| 15 |
+
def generate_text(input_prompt="", max_len=200, top_k=10, temp=0.5, top_p=0.95):
|
| 16 |
print('inputs: ')
|
| 17 |
print('prompt:', prompt)
|
| 18 |
print('max_len:', max_len)
|