updated model name
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ if "model_name" not in st.session_state:
|
|
37 |
|
38 |
@st.cache_resource
|
39 |
def get_pipeline(model_name):
|
40 |
-
model_name = "
|
41 |
device = 0 if torch.cuda.is_available() else -1
|
42 |
# if True or model_name == "meta-llama/Llama-2-7b-chat-hf" or model_name == "mistralai/Mistral-7B-Instruct-v0.2":
|
43 |
# chatbot = pipeline(model=model_name, task="conversational", device=device)#, model_kwargs=model_kwargs)
|
|
|
37 |
|
38 |
@st.cache_resource
|
39 |
def get_pipeline(model_name):
|
40 |
+
model_name = "gpt2-medium"
|
41 |
device = 0 if torch.cuda.is_available() else -1
|
42 |
# if True or model_name == "meta-llama/Llama-2-7b-chat-hf" or model_name == "mistralai/Mistral-7B-Instruct-v0.2":
|
43 |
# chatbot = pipeline(model=model_name, task="conversational", device=device)#, model_kwargs=model_kwargs)
|