Update space

app.py

@@ -7,7 +7,7 @@ For more information on `huggingface_hub` Inference API support, please check th
 # client = InferenceClient("karanzrk/bert-Causal-QA")
 
 from transformers import pipeline
-generator = pipeline('text-generation', model = 'karanzrk/bert-Causal-QA', tokenizer="bert-base-uncased")
+generator = pipeline('text-generation', model = 'karanzrk/bert-Causal-QA', tokenizer="bert-base-uncased", max_length = 128)
 # generator("Hello, I'm a language model", max_length = 30, num_return_sequences=3)
 
 
@@ -64,6 +64,7 @@ def inference(text):
     # classifier = pipeline("text-classification", model="karanzrk/essayl0")
     text = "Question: " + text
     output = generator(text)
+    answer = output["generated_text"]
     return output
 
 # launcher = gr.Interface(