acecalisto3 commited on
Commit
2b7f3e0
·
verified ·
1 Parent(s): 117c9ac

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -15,8 +15,7 @@ class JSONDecodeError(Exception):
15
  pass
16
 
17
  def generate_project_config(project_description):
18
- generator = pipeline('text-generation', model='gpt2')
19
-
20
  prompt = f"Generate a project configuration for the following project description: {project_description}\n\n{{"
21
 
22
  try:
 
15
  pass
16
 
17
  def generate_project_config(project_description):
18
+ generator = pipeline('text-generation', model='distilgpt2', device=-1)
 
19
  prompt = f"Generate a project configuration for the following project description: {project_description}\n\n{{"
20
 
21
  try: