Adding new error message for empty temperature sampling and repetition penalty

app.py

@@ -26,8 +26,9 @@ def single_generation(text,min_length,max_length,temperature,top_k,top_p,num_bea
     error_minmax=ValueError(f"ERROR: min length must be lower than or equal to max length! Given min length = {min_length}")
     error_numbeams_type=TypeError(f"ERROR: number of beams must be an integer not {type(num_beams)}")
     error_topk_type=TypeError(f"ERROR: top k must be an integer not {type(top_k)}")
-    error_minmax_type=TypeError(f"ERROR: min length and max length must be an integer not {type(min_length)} and {type(max_length)}")   
-    error_empty=ValueError("ERROR: Input Text cannot be empty!")
+    error_minmax_type=TypeError(f"ERROR: min length and max length must be an integer not {type(min_length)} and {type(max_length)}")
+    error_empty_temprep=TypeError("ERROR: temperature and repetition penalty cannot be zero!")
+    error_empty_text=ValueError("ERROR: Input Text cannot be empty!")
     error_unknown=TypeError("Unknown Error.")
 
     if text != '':
@@ -37,18 +38,21 @@ def single_generation(text,min_length,max_length,temperature,top_k,top_p,num_bea
                     if min_length <= max_length:
                         if temperature > 0:
                             if repetition_penalty >= 1:
-                                result = generator(text,
-                                                   min_length=min_length,
-                                                   max_length=max_length,
-                                                   temperature=temperature,
-                                                   top_k=top_k,
-                                                   top_p=top_p,
-                                                   num_beams=num_beams,
-                                                   repetition_penalty=repetition_penalty,
-                                                   do_sample=do_sample,
-                                                   no_repeat_ngram_size=2,
-                                                   num_return_sequences=1)
-                                return result[0]["generated_text"]
+                                if temperature AND repetition_penalty !='':
+                                    result = generator(text,
+                                                       min_length=min_length,
+                                                       max_length=max_length,
+                                                       temperature=temperature,
+                                                       top_k=top_k,
+                                                       top_p=top_p,
+                                                       num_beams=num_beams,
+                                                       repetition_penalty=repetition_penalty,
+                                                       do_sample=do_sample,
+                                                       no_repeat_ngram_size=2,
+                                                       num_return_sequences=1)
+                                    return result[0]["generated_text"]
+                                elif temperature AND repetition_penalty =='':
+                                    return error_empty_temprep
                             elif repetition_penalty < 1:
                                 return error_rep
                         elif temperature <= 0:
@@ -62,7 +66,7 @@ def single_generation(text,min_length,max_length,temperature,top_k,top_p,num_bea
         elif type(min_length) != int or type(max_length) != int:
             return error_minmax_type  
     elif text == '':
-        return error_empty
+        return error_empty_text
     else:
         return error_unknown