Update app.py

app.py

@@ -2,8 +2,9 @@ import gradio as gr
 from transformers import pipeline
 
 # Code Generation Pipeline
-code_pipeline = pipeline("text-generation", model="Salesforce/codegen-350M-mono", device=-1)
+code_pipeline = pipeline("text-generation", model="Salesforce/codegen-350M-mono", device=0)
 
+@spaces.GPU
 def generate_code(prompt):
     output = code_pipeline(prompt, max_length=50)
     return output[0]['generated_text']