Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,6 @@ import os
|
|
2 |
import streamlit as st
|
3 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
4 |
|
5 |
-
os.environ["HUGGINGFACE_HUB_API_TOKEN"] = os.getenv("HUGGINGFACE_HUB_API_TOKEN")
|
6 |
# Create a class for the session state
|
7 |
class SessionState:
|
8 |
def __init__(self):
|
@@ -19,7 +18,7 @@ temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.7)
|
|
19 |
|
20 |
# Load the model and tokenizer with a loading message
|
21 |
with st.spinner('Wait for model to respond..'):
|
22 |
-
model_name = "
|
23 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
24 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
25 |
|
|
|
2 |
import streamlit as st
|
3 |
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
4 |
|
|
|
5 |
# Create a class for the session state
|
6 |
class SessionState:
|
7 |
def __init__(self):
|
|
|
18 |
|
19 |
# Load the model and tokenizer with a loading message
|
20 |
with st.spinner('Wait for model to respond..'):
|
21 |
+
model_name = "intfloat/e5-mistral-7b-instruct"
|
22 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
23 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
24 |
|