sanchit-gandhi HF staff commited on
Commit
ede6cee
1 Parent(s): 33669c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -57,7 +57,7 @@ def query(payload):
57
  return response.json()
58
 
59
 
60
- def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
61
  payload = {
62
  "prompt": input_sentence,
63
  "do_sample": True,
@@ -68,7 +68,8 @@ def inference(input_sentence, max_length, sample_or_greedy, raw_text=False):
68
  payload
69
  )
70
 
71
- if raw_text:
 
72
  return None, data[0]['generated_text']
73
 
74
  width, height = 3326, 3326
 
57
  return response.json()
58
 
59
 
60
+ def inference(input_sentence, max_length, sample_or_greedy, raw_text=True):
61
  payload = {
62
  "prompt": input_sentence,
63
  "do_sample": True,
 
68
  payload
69
  )
70
 
71
+ # if raw_text:
72
+ if True:
73
  return None, data[0]['generated_text']
74
 
75
  width, height = 3326, 3326