pragneshbarik commited on
Commit
ab843c1
1 Parent(s): 6c6516f

improved dependancies

Browse files
Files changed (2) hide show
  1. app.py +0 -1
  2. utils.py +1 -0
app.py CHANGED
@@ -1,5 +1,4 @@
1
  import streamlit as st
2
- from utils import generate_text_embeddings
3
  from mistral7b import mistral
4
 
5
  import time
 
1
  import streamlit as st
 
2
  from mistral7b import mistral
3
 
4
  import time
utils.py CHANGED
@@ -8,6 +8,7 @@ API_TOKEN = os.getenv('HF_TOKEN')
8
  API_URL = "https://api-inference.huggingface.co/models/all-distilroberta-v1"
9
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
10
 
 
11
  def generate_text_embeddings(texts):
12
  model = SentenceTransformer('all-distilroberta-v1')
13
 
 
8
  API_URL = "https://api-inference.huggingface.co/models/all-distilroberta-v1"
9
  headers = {"Authorization": f"Bearer {API_TOKEN}"}
10
 
11
+
12
  def generate_text_embeddings(texts):
13
  model = SentenceTransformer('all-distilroberta-v1')
14