anugrahap commited on
Commit
2bf3482
·
1 Parent(s): 027f46b

Revised the error message

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -21,15 +21,15 @@ generator = pipeline('text-generation', model=model, tokenizer=tokenizer)
21
  # create the decoder parameter to generate the text
22
  def single_generation(text,min_length,max_length,temperature,top_k,top_p,num_beams,repetition_penalty,do_sample):
23
  # create local variable for error parameter
24
- error_rep=ValueError(f"ERROR: repetition penalty cannot be lower than one! Given rep penalty = {repetition_penalty}")
25
- error_temp=ValueError(f"ERROR: temperature cannot be zero or lower! Given temperature = {temperature}")
26
- error_minmax=ValueError(f"ERROR: min length must be lower than or equal to max length! Given min length = {min_length}")
27
- error_numbeams_type=TypeError(f"ERROR: number of beams must be an integer not {type(num_beams)}")
28
- error_topk_type=TypeError(f"ERROR: top k must be an integer not {type(top_k)}")
29
- error_minmax_type=TypeError(f"ERROR: min length and max length must be an integer not {type(min_length)} and {type(max_length)}")
30
- error_empty_temprep=TypeError("ERROR: temperature and repetition penalty cannot be empty!")
31
- error_empty_text=ValueError("ERROR: Input Text cannot be empty!")
32
- error_unknown=TypeError("Unknown Error.")
33
 
34
  if text != '':
35
  if type(min_length) == int and type(max_length) == int:
@@ -51,7 +51,7 @@ def single_generation(text,min_length,max_length,temperature,top_k,top_p,num_bea
51
  no_repeat_ngram_size=2,
52
  num_return_sequences=1)
53
  return result[0]["generated_text"]
54
- elif temperature and repetition_penalty is None:
55
  return error_empty_temprep
56
  elif repetition_penalty < 1:
57
  return error_rep
 
21
  # create the decoder parameter to generate the text
22
  def single_generation(text,min_length,max_length,temperature,top_k,top_p,num_beams,repetition_penalty,do_sample):
23
  # create local variable for error parameter
24
+ error_rep=gr.Error(f"ERROR: repetition penalty cannot be lower than one! Given rep penalty = {repetition_penalty}")
25
+ error_temp=gr.Error(f"ERROR: temperature cannot be zero or lower! Given temperature = {temperature}")
26
+ error_minmax=gr.Error(f"ERROR: min length must be lower than or equal to max length! Given min length = {min_length}")
27
+ error_numbeams_type=gr.Error(f"ERROR: number of beams must be an integer not {type(num_beams)}")
28
+ error_topk_type=gr.Error(f"ERROR: top k must be an integer not {type(top_k)}")
29
+ error_minmax_type=gr.Error(f"ERROR: min length and max length must be an integer not {type(min_length)} and {type(max_length)}")
30
+ error_empty_temprep=gr.Error("ERROR: temperature and repetition penalty cannot be empty!")
31
+ error_empty_text=gr.Error("ERROR: Input Text cannot be empty!")
32
+ error_unknown=gr.Error("Unknown Error.")
33
 
34
  if text != '':
35
  if type(min_length) == int and type(max_length) == int:
 
51
  no_repeat_ngram_size=2,
52
  num_return_sequences=1)
53
  return result[0]["generated_text"]
54
+ elif temperature or repetition_penalty is None:
55
  return error_empty_temprep
56
  elif repetition_penalty < 1:
57
  return error_rep