Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ def img2text(url):
|
|
| 8 |
|
| 9 |
def text2story(text):
|
| 10 |
story_generator = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B", device_map="auto", return_full_text=False)
|
| 11 |
-
prompt = f"Generate a short story under 100 words about {text}."
|
| 12 |
generated = story_generator(prompt, max_new_tokens=130, do_sample=True, temperature=0.7)
|
| 13 |
story_text = generated[0]['generated_text'].replace(prompt, "").strip()
|
| 14 |
return story_text
|
|
|
|
| 8 |
|
| 9 |
def text2story(text):
|
| 10 |
story_generator = pipeline("text-generation", model="Qwen/Qwen2.5-0.5B", device_map="auto", return_full_text=False)
|
| 11 |
+
prompt = f"Generate a short story under 100 words about {text} without other narrative."
|
| 12 |
generated = story_generator(prompt, max_new_tokens=130, do_sample=True, temperature=0.7)
|
| 13 |
story_text = generated[0]['generated_text'].replace(prompt, "").strip()
|
| 14 |
return story_text
|