Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,6 +1,10 @@
 import streamlit as st
 import transformers
 from transformers import pipeline
+import os
+
+# Read token from environment (set as secret in Space settings)
+token = os.environ.get("token")
 
 def infer(sent, max_length, num_return_sequences):
     generator = pipeline('text-generation', model='gpt2')
@@ -8,12 +12,12 @@ def infer(sent, max_length, num_return_sequences):
     return generator(sent, max_length=max_length, num_return_sequences=num_return_sequences)
 
 def instantiate_gpt2(max_length_ : int, num_return_sequences : int, text : str) -> dict:
-    pipe = pipeline(task='text-generation', model='Iscte-Sintra/GPT2-Kriolu')
+    pipe = pipeline(task='text-generation', model='Iscte-Sintra/GPT2-Kriolu', token=token)
     results = pipe(text, max_length=max_length_, num_return_sequences=num_return_sequences)
     return results
 
 def instantiate_roberta(top_k : int, text : str) -> dict:
-    pipe = pipeline("fill-mask", model="Iscte-Sintra/RoBERTa-Kriolu")
+    pipe = pipeline("fill-mask", model="Iscte-Sintra/RoBERTa-Kriolu", token=token)
     return pipe(text, top_k=top_k)
 
 def build_gpt2_page():