Update app.py
Browse files
app.py
CHANGED
|
@@ -15,7 +15,8 @@ from transformers import pipeline
|
|
| 15 |
|
| 16 |
def generate_story(caption):
|
| 17 |
# 使用文本生成 pipeline
|
| 18 |
-
generator = pipeline("text-generation", model="gpt2")
|
|
|
|
| 19 |
prompt = f"由以下图片得到的描述: '{caption}'，请根据这个描述生成一个完整的童话故事，故事至少100个单词。"
|
| 20 |
result = generator(prompt, max_length=300, num_return_sequences=1)
|
| 21 |
story = result[0]['generated_text']
|
|
@@ -35,7 +36,7 @@ from PIL import Image
|
|
| 35 |
|
| 36 |
def main():
|
| 37 |
st.title("儿童故事生成应用")
|
| 38 |
-
st.write("
|
| 39 |
|
| 40 |
uploaded_file = st.file_uploader("选择一张图片", type=["png", "jpg", "jpeg"])
|
| 41 |
if uploaded_file is not None:
|
|
|
|
| 15 |
|
| 16 |
def generate_story(caption):
|
| 17 |
# 使用文本生成 pipeline
|
| 18 |
+
generator = pipeline("text-generation", model="openai-community/gpt2")
|
| 19 |
+
# Use a pipeline as a high-level helper
|
| 20 |
prompt = f"由以下图片得到的描述: '{caption}'，请根据这个描述生成一个完整的童话故事，故事至少100个单词。"
|
| 21 |
result = generator(prompt, max_length=300, num_return_sequences=1)
|
| 22 |
story = result[0]['generated_text']
|
|
|
|
| 36 |
|
| 37 |
def main():
|
| 38 |
st.title("儿童故事生成应用")
|
| 39 |
+
st.write("上传一张图片，我们将根据图片生成有趣的故事，并转换成语音播放给你听！!!")
|
| 40 |
|
| 41 |
uploaded_file = st.file_uploader("选择一张图片", type=["png", "jpg", "jpeg"])
|
| 42 |
if uploaded_file is not None:
|