sshenai commited on
Commit
f17e48b
·
verified ·
1 Parent(s): 4505c1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -1,3 +1,12 @@
 
 
 
 
 
 
 
 
 
1
  # 加载模型(全局加载提升性能)
2
  MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
3
  SUMMARIZER = pipeline("summarization", model="facebook/bart-large-cnn")
 
1
+ from datasets import load_dataset
2
+ import numpy as np
3
+ from transformers import AutoTokenizer, AutoModel, pipeline
4
+ import torch
5
+ import torch.nn.functional as F
6
+ from sentence_transformers import SentenceTransformer
7
+ from sklearn.metrics.pairwise import cosine_similarity
8
+ import gradio as gr
9
+
10
  # 加载模型(全局加载提升性能)
11
  MODEL_NAME = "sentence-transformers/all-MiniLM-L6-v2"
12
  SUMMARIZER = pipeline("summarization", model="facebook/bart-large-cnn")