acecalisto3 commited on
Commit
73fc0ce
·
verified ·
1 Parent(s): ee8457c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -11,14 +11,14 @@ import importlib
11
  import os
12
  from transformers import AutoModelForSequenceClassification, pipeline, AutoTokenizer
13
 
14
- # Load the tokenizer explicitly
15
- tokenizer = AutoTokenizer.from_pretrained("microsoft/CodeGPT-small-py", clean_up_tokenization_spaces=True)
16
 
17
- # Initialize the model
18
- model = AutoModelForSequenceClassification.from_pretrained("microsoft/CodeGPT-small-py") # Use a public model
19
 
20
- # Initialize the pipeline
21
- code_generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
22
 
23
  # Initialize Flask app
24
  app = Flask(__name__)
 
11
  import os
12
  from transformers import AutoModelForSequenceClassification, pipeline, AutoTokenizer
13
 
14
+ # Load the tokenizer explicitly
15
+ tokenizer = AutoTokenizer.from_pretrained("microsoft/CodeGPT-small-py", clean_up_tokenization_spaces=True)
16
 
17
+ # Initialize the model
18
+ model = AutoModelForSequenceClassification.from_pretrained("microsoft/CodeGPT-small-py") # Use a public model
19
 
20
+ # Initialize the pipeline
21
+ code_generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
22
 
23
  # Initialize Flask app
24
  app = Flask(__name__)