Spaces:
Sleeping
Sleeping
Update server.py
Browse files
server.py
CHANGED
|
@@ -8,7 +8,7 @@ import os
|
|
| 8 |
class TopicBanningAPI(ls.LitAPI):
|
| 9 |
def setup(self, device):
|
| 10 |
model_name = "facebook/bart-large-mnli"
|
| 11 |
-
self.pipeline = pipeline("zero-shot-classification", model=
|
| 12 |
|
| 13 |
def decode_request(self, request):
|
| 14 |
return request
|
|
|
|
| 8 |
class TopicBanningAPI(ls.LitAPI):
|
| 9 |
def setup(self, device):
|
| 10 |
model_name = "facebook/bart-large-mnli"
|
| 11 |
+
self.pipeline = pipeline("zero-shot-classification", model=model_name, token=os.getenv("HF_TOKEN"))
|
| 12 |
|
| 13 |
def decode_request(self, request):
|
| 14 |
return request
|