zaidmehdi commited on
Commit
4b51583
1 Parent(s): 9a23b5c

fix indentation

Browse files
Files changed (1) hide show
  1. src/main.py +2 -2
src/main.py CHANGED
@@ -8,7 +8,7 @@ from utils import extract_hidden_state
8
  app = Flask(__name__)
9
 
10
  with open("../models/logistic_regression.pkl", "rb") as f:
11
- model = pickle.load(f)
12
 
13
  model_name = "moussaKam/AraBART"
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
@@ -28,7 +28,7 @@ def classify_arabic_dialect():
28
 
29
  return jsonify({"class": predicted_class}), 200
30
  except Exception as e:
31
- return jsonify({"error": str(e)}), 500
32
 
33
 
34
  def main():
 
8
  app = Flask(__name__)
9
 
10
  with open("../models/logistic_regression.pkl", "rb") as f:
11
+ model = pickle.load(f)
12
 
13
  model_name = "moussaKam/AraBART"
14
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
28
 
29
  return jsonify({"class": predicted_class}), 200
30
  except Exception as e:
31
+ return jsonify({"error": str(e)}), 500
32
 
33
 
34
  def main():