Update app.py
Browse files
app.py
CHANGED
|
@@ -4,7 +4,7 @@ import torch
|
|
| 4 |
|
| 5 |
access_token = st.secrets["access_token"]
|
| 6 |
|
| 7 |
-
pipe = pipeline(model="
|
| 8 |
text = st.text_area("Enter a sentence !!!")
|
| 9 |
|
| 10 |
if text:
|
|
|
|
| 4 |
|
| 5 |
access_token = st.secrets["access_token"]
|
| 6 |
|
| 7 |
+
pipe = pipeline(model="distilbert/distilbert-base-uncased", torch_dtype=torch.bfloat16, device_map="auto", token=access_token)
|
| 8 |
text = st.text_area("Enter a sentence !!!")
|
| 9 |
|
| 10 |
if text:
|