karanzrk commited on
Commit
03a21d4
·
1 Parent(s): 36833e9

Update space

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ For more information on `huggingface_hub` Inference API support, please check th
7
  # client = InferenceClient("karanzrk/bert-Causal-QA")
8
 
9
  from transformers import pipeline
10
- generator = pipeline('text-generation', model = 'karanzrk/bert-Causal-QA')
11
  # generator("Hello, I'm a language model", max_length = 30, num_return_sequences=3)
12
 
13
 
 
7
  # client = InferenceClient("karanzrk/bert-Causal-QA")
8
 
9
  from transformers import pipeline
10
+ generator = pipeline('text-generation', model = 'karanzrk/bert-Causal-QA', tokenizer="bert-base-uncased")
11
  # generator("Hello, I'm a language model", max_length = 30, num_return_sequences=3)
12
 
13