Spaces:
Running
Running
pragneshbarik
commited on
Commit
•
ab843c1
1
Parent(s):
6c6516f
improved dependancies
Browse files
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 |
|