Update generation parameters

app.py

@@ -2,14 +2,18 @@ import gradio as gr
 from transformers import pipeline
 
 # Load the Hugging Face model
-# Replace 'your-username/your-model-name' with the actual path to your Hugging Face model
 model_path = "patrixtano/mt5-small-anaphora_czech_6e"
 model_pipeline = pipeline("text2text-generation", model=model_path)
-
+tokenizer = AutoTokenizer.from_pretrained("patrixtano/mt5-small-anaphora_czech_3e")
 def predict(text_input):
     """
     Generate a prediction for the given input text using the Hugging Face model.
     """
+    input_length = len(tokenizer(input_text)["input_ids"])
+    generation_parameters = {
+    "min_length": input_length + 5,  # Set your desired minimum length
+    "max_length": input_length + 10  # Set your desired maximum length
+    }
     try:
         result = model_pipeline(text_input)
         # Extract and return the generated text