From 227ce592935827658b3c8adc11cc728979265238 Mon Sep 17 00:00:00 2001
From: Mehrdad Farahani <m3hrdadfi@gmail.com>
Date: Thu, 12 Aug 2021 17:39:44 +0200
Subject: [PATCH] Fix permission

---
 apps/st_app/app.py | 17 +++++++++++------
 1 file changed, 11 insertions(+), 6 deletions(-)

diff --git a/apps/st_app/app.py b/apps/st_app/app.py
index fd610d8..111f033 100644
--- a/apps/st_app/app.py
+++ b/apps/st_app/app.py
@@ -1,5 +1,5 @@
 import streamlit as st
-
+import os
 import torch
 from transformers import pipeline
 from transformers import AutoConfig, AutoTokenizer, AutoModelForTokenClassification
@@ -16,6 +16,7 @@
     "Persian (fa)": "m3hrdadfi/typo-detector-distilbert-fa",
     "Icelandic (is)": "m3hrdadfi/typo-detector-distilbert-is",
 }
+API_TOKEN = os.environ.get("API_TOKEN")
 
 
 class TypoDetector:
@@ -34,11 +35,15 @@ def __init__(
         self.nlp = None
         self.normalizer = None
 
-    def load(self):
+    def load(self, api_token=None):
+        api_token = api_token if api_token else False
         if not self.debug:
-            self.config = AutoConfig.from_pretrained(self.model_name_or_path)
-            self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path)
-            self.model = AutoModelForTokenClassification.from_pretrained(self.model_name_or_path, config=self.config)
+            self.config = AutoConfig.from_pretrained(self.model_name_or_path, use_auth_token=api_token)
+            self.tokenizer = AutoTokenizer.from_pretrained(self.model_name_or_path, use_auth_token=api_token)
+            self.model = AutoModelForTokenClassification.from_pretrained(
+                self.model_name_or_path,
+                config=self.config,
+                use_auth_token=api_token)
             self.nlp = pipeline(
                 self.task_name,
                 model=self.model,
@@ -70,7 +75,7 @@ def load_typo_detectors():
     is_detector.load()
 
     fa_detector = TypoDetector(MODELS["Persian (fa)"])
-    fa_detector.load()
+    fa_detector.load(api_token=API_TOKEN)
 
     return {
         "en": en_detector,