ElPremOoO commited on
Commit
53b8b52
·
verified ·
1 Parent(s): 9e48d61

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -4,7 +4,7 @@ from transformers import RobertaTokenizer
4
  import os
5
 
6
  # Initialize Flask app
7
- app = Flask(name)
8
 
9
  # Load the trained model and tokenizer
10
  tokenizer = RobertaTokenizer.from_pretrained("microsoft/codebert-base")
@@ -54,5 +54,5 @@ def predict():
54
 
55
 
56
  # Run the Flask app
57
- if name == "main":
58
  app.run(debug=True)
 
4
  import os
5
 
6
  # Initialize Flask app
7
+ app = Flask(__name__)
8
 
9
  # Load the trained model and tokenizer
10
  tokenizer = RobertaTokenizer.from_pretrained("microsoft/codebert-base")
 
54
 
55
 
56
  # Run the Flask app
57
+ if __name__ == "main":
58
  app.run(debug=True)