Spaces:
Runtime error
Runtime error
Revised the error message
Browse files
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=
|
25 |
-
error_temp=
|
26 |
-
error_minmax=
|
27 |
-
error_numbeams_type=
|
28 |
-
error_topk_type=
|
29 |
-
error_minmax_type=
|
30 |
-
error_empty_temprep=
|
31 |
-
error_empty_text=
|
32 |
-
error_unknown=
|
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
|
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
|