lmastm commited on
Commit
3e16169
·
verified ·
1 Parent(s): 4389662

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
 
5
  access_token = st.secrets["access_token"]
6
 
7
- pipe = pipeline(model="Intel/neural-chat-7b-v3-1", torch_dtype=torch.bfloat16, device_map="auto", token=access_token)
8
  text = st.text_area("Enter a sentence !!!")
9
 
10
  if text:
 
4
 
5
  access_token = st.secrets["access_token"]
6
 
7
+ pipe = pipeline("text-classification", token=access_token)
8
  text = st.text_area("Enter a sentence !!!")
9
 
10
  if text: