Update app.py

app.py

@@ -1,9 +1,7 @@
 from transformers import pipeline
-import torch
 import gradio as gr
 
-# Make sure torch is imported and available
-generator = pipeline("text2text-generation", model="google/flan-t5-base")
+generator = pipeline("text-generation", model="Salesforce/codegen2-1B")
 
 def generate_code(prompt):
     output = generator(prompt, max_length=2048, do_sample=True)[0]["generated_text"]