Update app.py

app.py

@@ -1,9 +1,17 @@
+import os
 from flask import Flask, request, jsonify
 from transformers import pipeline
 from langdetect import detect
+from huggingface_hub import login
 
 app = Flask(__name__)
 
+# Gets the Token from secrect
+hf_hub_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
+
+# Logging in
+login(token=hf_hub_token)
+
 # Load models
 hebrew_model = pipeline("text-generation", model="onlplab/alephbert-base")
 english_model = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")