reydeuss commited on
Commit
1a152ce
·
verified ·
1 Parent(s): 671de3c

tried authenticating with readonly token for adapter

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  # app.py for Gradio with PEFT
2
  import gradio as gr
3
  import torch
 
 
4
  from transformers import T5Tokenizer, T5ForConditionalGeneration
5
  from peft import PeftModel, PeftConfig
6
 
@@ -11,6 +13,9 @@ tokenizer = None
11
  def load_model_once():
12
  global model, tokenizer
13
  if model is None:
 
 
 
14
  # Load base model
15
  base_model_name = "cahya/t5-base-indonesian-summarization-cased"
16
  tokenizer = T5Tokenizer.from_pretrained(base_model_name)
 
1
  # app.py for Gradio with PEFT
2
  import gradio as gr
3
  import torch
4
+ import os
5
+ from huggingface_hub import login
6
  from transformers import T5Tokenizer, T5ForConditionalGeneration
7
  from peft import PeftModel, PeftConfig
8
 
 
13
  def load_model_once():
14
  global model, tokenizer
15
  if model is None:
16
+ hf_token = os.environ.get('HF_TOKEN')
17
+ login(token=hf_token)
18
+
19
  # Load base model
20
  base_model_name = "cahya/t5-base-indonesian-summarization-cased"
21
  tokenizer = T5Tokenizer.from_pretrained(base_model_name)