youssef1214 commited on
Commit
877dc8a
1 Parent(s): 163695f

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -154,7 +154,7 @@ def original_values(num):
154
  elif num == 6:
155
  return 'fear'
156
  def modelsummary(data):
157
- result = pipeline1(text,
158
  pad_token_id= tokenizer.eos_token_id,
159
  num_beams=4,
160
  repetition_penalty=3.0,
 
154
  elif num == 6:
155
  return 'fear'
156
  def modelsummary(data):
157
+ result = pipeline1(data,
158
  pad_token_id= tokenizer.eos_token_id,
159
  num_beams=4,
160
  repetition_penalty=3.0,