Update app.py
Browse files
app.py
CHANGED
|
@@ -48,11 +48,6 @@ def generate_story(name: str, hobby: str):
|
|
| 48 |
full_text = out[0]["generated_text"]
|
| 49 |
story = full_text.split("[Assistant]:")[-1].replace("</s>", "").strip()
|
| 50 |
return story
|
| 51 |
-
|
| 52 |
-
out = pipe(prompt, max_new_tokens=900, temperature=0.9, top_p=0.95, do_sample=True)
|
| 53 |
-
full_text = out[0]["generated_text"]
|
| 54 |
-
story = full_text.split("[Assistant]:")[-1].replace("</s>", "").strip()
|
| 55 |
-
return story
|
| 56 |
|
| 57 |
iface = gr.Interface(
|
| 58 |
fn=generate_story,
|
|
|
|
| 48 |
full_text = out[0]["generated_text"]
|
| 49 |
story = full_text.split("[Assistant]:")[-1].replace("</s>", "").strip()
|
| 50 |
return story
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 51 |
|
| 52 |
iface = gr.Interface(
|
| 53 |
fn=generate_story,
|