update prompt

app.py

@@ -1,6 +1,5 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed
-from transformers import pipeline
+from transformers import AutoTokenizer, AutoModelForCausalLM, set_seed, pipeline
 
 
 title = "Text to Python Converter"
@@ -22,7 +21,8 @@ def make_doctring(gen_prompt):
 def code_generation(gen_prompt, max_tokens, temperature=0.6, seed=42):
     set_seed(seed)
     pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
-    generated_text = pipe(make_doctring(gen_prompt), do_sample=True, top_p=0.95, temperature=temperature, max_new_tokens=max_tokens)[0]['generated_text']
+    prompt = make_doctring(gen_prompt)
+    generated_text = pipe(prompt, do_sample=True, top_p=0.95, temperature=temperature, max_new_tokens=max_tokens)[0]['generated_text']
     return generated_text