Nathan Gebreab commited on
Commit
7c41b8e
·
1 Parent(s): 5ae280b

added hf_token to secrets and fixed login

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -21,16 +21,18 @@ https://www.kaggle.com/datasets/joebeachcapital/30000-spotify-songs
21
  """
22
 
23
  import torch
 
24
  from transformers import pipeline
25
  from sentence_transformers import SentenceTransformer
26
  import pandas as pd
27
  import numpy as np
28
  import warnings
29
  import gradio as gr
30
- from huggingface_hub import InferenceClient
31
  from huggingface_hub import login
32
 
33
- login()
 
 
34
 
35
  model_id="meta-llama/Llama-3.2-3B-Instruct"
36
 
 
21
  """
22
 
23
  import torch
24
+ import os
25
  from transformers import pipeline
26
  from sentence_transformers import SentenceTransformer
27
  import pandas as pd
28
  import numpy as np
29
  import warnings
30
  import gradio as gr
 
31
  from huggingface_hub import login
32
 
33
+ hf_token = os.getenv("HF_TOKEN")
34
+ if hf_token:
35
+ login(token=hf_token)
36
 
37
  model_id="meta-llama/Llama-3.2-3B-Instruct"
38