JuliaWolken commited on
Commit
433732d
·
verified ·
1 Parent(s): a9f90bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -1,11 +1,10 @@
1
- # -*- coding: utf-8 -*-
2
- """Deploy to Gradio.ipynb
3
-
4
- Automatically generated by Colab.
5
-
6
- Original file is located at
7
- https://colab.research.google.com/drive/11013nKY0n_XCe50LERXzoqrHsKFwuRBl
8
- """
9
 
10
  chunks_data_path = 'WB_KB/chunks.csv'
11
  model = SentenceTransformer('JuliaWolken/fine_tuned_model_with_triplets')
 
1
+ import gradio as gr
2
+ import pandas as pd
3
+ from sentence_transformers import SentenceTransformer
4
+ from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
+ from sklearn.metrics.pairwise import cosine_similarity
6
+ import torch
7
+ import re
 
8
 
9
  chunks_data_path = 'WB_KB/chunks.csv'
10
  model = SentenceTransformer('JuliaWolken/fine_tuned_model_with_triplets')