prevent short length

app.py

@@ -7,10 +7,12 @@ model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", torch_dtype=torc
 
 def generate(prompt, length):
     inputs = tokenizer(prompt, return_tensors="pt", return_attention_mask=False)
+    if length < inputs.length:
+        length = inputs.length
     outputs = model.generate(**inputs, max_length=length)
     return tokenizer.batch_decode(outputs)[0]
 
-demo = gr.Interface(fn=generate, inputs=["text", "number"], outputs="text")
+demo = gr.Interface(fn=generate, inputs=["text", gr.Number(value=50, label="max length",maximum=200)], outputs="text")
 
 if __name__ == "__main__":
    demo.launch(show_api=False)