Update app.py

app.py

@@ -1,15 +1,22 @@
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import safetensors_rust
 
 # Load the trained model and tokenizer
-model_path = 'viv/AIKIA'  # Make sure this is correct, either local path or Hugging Face path
+model_path = 'viv/AIKIA'  # Ensure this path is correct, either local or Hugging Face path
 tokenizer = AutoTokenizer.from_pretrained("nlpaueb/bert-base-greek-uncased-v1")
-model = AutoModelForSequenceClassification.from_pretrained(model_path)
+
+# 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")
 
 # Preprocessing function for Greek text
 def preprocessing_greek(text):
-    text = text.lower()  # Example step
+    text = text.lower()  # Example step: Convert to lowercase
     return text
 
 # Prediction function