Update app.py

app.py

@@ -1,6 +1,17 @@
 import gradio as gr
 import random
 import time
+from transformers import AutoModelForQuestionAnswering, pipeline
+
+# 设置使用GPU
+device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# 初始化模型
+model_name = "bert-base-uncased"
+model = AutoModelForQuestionAnswering.from_pretrained(model_name).to(device)
+
+# 创建管道
+qa_pipeline = pipeline("question-answering", model=model, tokenizer=model_name)
 
 # 假设这是队标的URL或本地路径
 team_logo_url = './team-logo.jpg'  # 替换为你的队标图片路径
@@ -32,7 +43,6 @@ _我们诚挚地感谢您的支持与反馈。_
 _feedback email:hhyxnh@gmail_
 
 
-
 """
 
 with gr.Blocks() as demo: