Update app.py

app.py

@@ -31,6 +31,9 @@ def generate_code(gen_prompt, max_tokens, temperature=0.6, seed=42):
     generated_text = pipe(prompt, do_sample=True, top_p=0.95, temperature=temperature, max_new_tokens=max_tokens)[0]['generated_text']
     return generated_text
 
+def save_to_text_file(output_text):
+    with open("generated_code.txt", "w") as file:
+        file.write(output_text)
 
 iface = gr.Interface(
     fn=generate_code,