Update gri.py

gri.py

@@ -3,10 +3,21 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 from deep_translator import GoogleTranslator
 from langdetect import detect
 import torch
+import os
 
+MODEL_NAME = "cardiffnlp/twitter-roberta-base-sentiment"
 MODEL_DIR = "model"
-tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
-model = AutoModelForSequenceClassification.from_pretrained(MODEL_DIR, from_safetensors=True)
+
+# Download model if not present
+if not os.path.exists(MODEL_DIR) or not os.listdir(MODEL_DIR):
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+    tokenizer.save_pretrained(MODEL_DIR)
+    model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
+    model.save_pretrained(MODEL_DIR)
+else:
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_DIR)
+    model = AutoModelForSequenceClassification.from_pretrained(MODEL_DIR)
+
 emotion_labels = {
     0: "Negative 😕",
     1: "Neutral 😐",
@@ -39,4 +50,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch(share = True)  # Set share=True to allow public access
+    iface.launch(share=False)