geethareddy commited on
Commit
6294632
·
verified ·
1 Parent(s): ef9691a

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +4 -0
config.py ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ class Config:
2
+ MODEL_NAME = "distilbert-base-uncased" # Model name (can be replaced with another model)
3
+ FLASK_PORT = 5000
4
+ FLASK_HOST = "0.0.0.0"