karanzrk commited on
Commit
c547b89
1 Parent(s): 3db5eda

changed runner

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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', tokenizer="bert-base-uncased", max_length = 128)
11
  # generator("Hello, I'm a language model", max_length = 30, num_return_sequences=3)
12
 
13
 
@@ -76,7 +76,7 @@ def inference(text):
76
  with gr.Blocks() as demo:
77
  gr.Markdown(
78
  """
79
- # Welcome to bert-demo
80
  Ask your question
81
  """
82
  )
 
7
  # client = InferenceClient("karanzrk/bert-Causal-QA")
8
 
9
  from transformers import pipeline
10
+ generator = pipeline('text-generation', model = 'karanzrk/gpt2-qa', tokenizer="gpt2", max_length = 128)
11
  # generator("Hello, I'm a language model", max_length = 30, num_return_sequences=3)
12
 
13
 
 
76
  with gr.Blocks() as demo:
77
  gr.Markdown(
78
  """
79
+ # Welcome to gpt2-demo
80
  Ask your question
81
  """
82
  )