GokulRajaR commited on
Commit
0c25e5d
·
verified ·
1 Parent(s): 3b75a21

Update server.py

Browse files
Files changed (1) hide show
  1. server.py +1 -1
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=model, token=os.getenv("HF_TOKEN"))
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