Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from transformers import BertTokenizer, BertForSequenceClassification, BertConfig
+from transformers import BertTokenizer, BertForSequenceClassification
 import torch
 import pickle
 from collections import defaultdict
@@ -13,22 +13,14 @@ def load_model():
         label_encoder = pickle.load(f)
 
     # Tokenizer
-    tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
-
-    # Config
-    config = BertConfig.from_pretrained('bert-base-uncased', num_labels=len(label_encoder.classes_))
-
-    # Model
-    model = BertForSequenceClassification(config)
-    
-    # Safetensors faylını yükləyirik
-    state_dict = load_file("best_model/model.safetensors")
-    model.load_state_dict(state_dict)
+    tokenizer = BertTokenizer.from_pretrained("best_model")
 
+    # Model (safetensors avtomatik dəstəklənir)
+    model = BertForSequenceClassification.from_pretrained("best_model", use_safetensors=True)
     model.eval()
+
     return tokenizer, model, label_encoder
 
-    
 tokenizer, model, label_encoder = load_model()
 
 # Prediction funksiyası
@@ -73,4 +65,4 @@ iface = gr.Interface(
 
 # Launch
 if __name__ == "__main__":
-    iface.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)), debug=True)
+    iface.launch(server_name="0.0.0.0", server_port=int(os.environ.get("PORT", 7860)))