Spaces:
Sleeping
Sleeping
fix indentation
Browse files- 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 |
-
|
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 |
-
|
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():
|