NLP-Text-Summarization

Build error

Alexvatti commited on Feb 17, 2025

Commit

1451f0b

verified ·

1 Parent(s): 4dcf075

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import gradio as gr
 from transformers import pipeline
 # Code Generation Pipeline
-code_pipeline = pipeline("text-generation", model="Salesforce/codegen-350M-mono")
 def generate_code(prompt):
     output = code_pipeline(prompt, max_length=50)

 from transformers import pipeline
 # Code Generation Pipeline
+code_pipeline = pipeline("text-generation", model="Salesforce/codegen-350M-mono", device=-1)
 def generate_code(prompt):
     output = code_pipeline(prompt, max_length=50)