Update app.py

app.py

@@ -1,18 +1,13 @@
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
-import safetensors_rust
 
 # Load the trained model and tokenizer
-model_path = 'viv/AIKIA'  # Ensure this path is correct, either local or Hugging Face path
+model_path = 'viv/AIKIA/pytorch_model.bin'  # Path to the .bin file
 tokenizer = AutoTokenizer.from_pretrained("nlpaueb/bert-base-greek-uncased-v1")
 
-# Try loading the model, fallback to `.bin` if `.safetensors` fails
-try:
-    model = AutoModelForSequenceClassification.from_pretrained(model_path)
-except safetensors_rust.SafetensorError:
-    print("Safetensors failed, trying to load bin file.")
-    model = AutoModelForSequenceClassification.from_pretrained("viv/AIKIA/pytorch_model.bin")
+# Directly load the model from the .bin file
+model = AutoModelForSequenceClassification.from_pretrained(model_path, from_tf=False, config='viv/AIKIA/config.json')
 
 # Preprocessing function for Greek text
 def preprocessing_greek(text):