ZEROTSUDIOS commited on
Commit
62868d6
·
verified ·
1 Parent(s): cb9f4da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -1,4 +1,4 @@
1
- from flask import Flask, request, jsonify
2
  from flask_cors import CORS
3
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
4
  from sentence_transformers import SentenceTransformer
@@ -109,10 +109,10 @@ def load_ood_thresholds(model_path):
109
  def load_intent_resources():
110
  global intent_model, intent_tokenizer, intent_classes, intent_thresholds
111
  try:
112
- print("⏳ Loading intent model...")
113
- intent_model = AutoModelForSequenceClassification.from_pretrained("ZEROTSUDIOS/Bipa-Classification")
114
  print("✅ Model loaded")
115
- intent_tokenizer = AutoTokenizer.from_pretrained("ZEROTSUDIOS/Bipa-Classification")
116
  print("✅ Tokenizer loaded")
117
  with open(os.path.join(INTENT_MODEL_PATH, "intent_classes.pkl"), "rb") as f:
118
  intent_classes = pickle.load(f)
 
1
+ ""from flask import Flask, request, jsonify
2
  from flask_cors import CORS
3
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
4
  from sentence_transformers import SentenceTransformer
 
109
  def load_intent_resources():
110
  global intent_model, intent_tokenizer, intent_classes, intent_thresholds
111
  try:
112
+ print("⏳ Loading local intent model from model/")
113
+ intent_model = AutoModelForSequenceClassification.from_pretrained(INTENT_MODEL_PATH)
114
  print("✅ Model loaded")
115
+ intent_tokenizer = AutoTokenizer.from_pretrained(INTENT_MODEL_PATH)
116
  print("✅ Tokenizer loaded")
117
  with open(os.path.join(INTENT_MODEL_PATH, "intent_classes.pkl"), "rb") as f:
118
  intent_classes = pickle.load(f)