Update app.py

app.py

@@ -19,6 +19,12 @@ def generate2(prompt, max_new_tokens):
     generated_ids = model2.generate(batch["input_ids"], max_new_tokens=150)
     output = tokenizer2.batch_decode(generated_ids, skip_special_tokens=True)
     return output[0]
+def generate2_test(prompt):
+    batch = tokenizer2(prompt, return_tensors="pt")
+    generated_ids = model2.generate(batch["input_ids"], max_new_tokens=150)
+    output = tokenizer2.batch_decode(generated_ids, skip_special_tokens=True)
+    return output[0]
+    
 def generate_prompt(type, prompt, max_new_tokens):
     if type==1:
         return generate(prompt)
@@ -29,4 +35,4 @@ input_component = gr.Textbox(label = "Input a persona, e.g. photographer", value
 output_component = gr.Textbox(label = "Prompt")
 examples = [["photographer"], ["developer"]]
 description = ""
-gr.Interface(generate, inputs = input_component, outputs=output_component, examples=examples, title = "👨🏻‍🎤 ChatGPT Prompt Generator v12 👨🏻‍🎤", description=description).launch()
+gr.Interface(generate2_test, inputs = input_component, outputs=output_component, examples=examples, title = "👨🏻‍🎤 ChatGPT Prompt Generator v12 👨🏻‍🎤", description=description).launch()