akshaytrikha commited on
Commit
2ac3c8d
1 Parent(s): f12af95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ trump = pipeline("text-generation", model=model, tokenizer=tokenizer, config={"m
18
 
19
  def generate(text):
20
  result = trump(text, num_return_sequences=1)
21
- return result[0]["generated_text"]
22
 
23
  examples = [
24
  ["Today I'll be"],
 
18
 
19
  def generate(text):
20
  result = trump(text, num_return_sequences=1)
21
+ return result[0]["generated_text"].replace('"', '') # remove quotation marks
22
 
23
  examples = [
24
  ["Today I'll be"],