Steven-GU-Yu-Di commited on
Commit
84cc85e
·
verified ·
1 Parent(s): 6cad4f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,8 +1,8 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- # 加载 Visual Question Answering 模型
5
- vqa_pipeline = pipeline("text2text-generation", model="mrm8488/t5-base-finetuned-vqa")
6
 
7
  # 加载文本到语音模型
8
  text_to_speech_pipeline = pipeline("text-to-speech", model="tts_model_name")
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ # 加载 Visual Question Answering 模型 microsoft/git-base-vqav2
5
+ vqa_pipeline = pipeline("text2text-generation", model="microsoft/git-base-vqav2")
6
 
7
  # 加载文本到语音模型
8
  text_to_speech_pipeline = pipeline("text-to-speech", model="tts_model_name")