Update app.py

app.py

@@ -31,9 +31,9 @@ def useZephyr(prompt):
     return outputs[0]["generated_text"]
 
 
-def generatePrompt(prompt, max_new_tokens):
+def generatePrompt(prompt):
     batch = tokenizer(prompt, return_tensors="pt")
-    generated_ids = model.generate(batch["input_ids"], max_new_tokens=max_new_tokens)
+    generated_ids = model.generate(batch["input_ids"])
     output = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)
     prompt = output[0]
 
@@ -55,13 +55,12 @@ def generate_prompt(prompt, max_new_tokens):
 
 
 input_prompt = gr.Textbox(label="Prompt", value="photographer")
-input_maxtokens = gr.Textbox(label="Max tokens", value="150")
 output_component = gr.Textbox(label="Output")
 examples = [["photographer"], ["developer"], ["teacher"], [
     "human resources staff"], ["recipe for ham croquettes"]]
 description = ""
 
-PerfectGPT = gr.Interface(generatePrompt, inputs=[input_prompt, input_maxtokens], outputs=output_component,
+PerfectGPT = gr.Interface(generatePrompt, inputs=input_prompt, outputs=output_component,
                           examples=examples, title="🗿 PerfectGPT v1 🗿", description=description)
 
 PerfectGPT.launch()