karths commited on
Commit
d964496
·
verified ·
1 Parent(s): a14201f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  import torch
4
  import numpy as np
5
  import random
6
- from huggingface_hub import login, HfFolder
7
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoModelForCausalLM, TextIteratorStreamer
8
  from scipy.special import softmax
9
  import logging
@@ -28,7 +28,6 @@ if torch.cuda.is_available():
28
 
29
  # Login to Hugging Face
30
  token = os.getenv("hf_token")
31
- HfFolder.save_token(token)
32
  login(token)
33
 
34
  model_paths = [
 
3
  import torch
4
  import numpy as np
5
  import random
6
+ from huggingface_hub import login
7
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, AutoModelForCausalLM, TextIteratorStreamer
8
  from scipy.special import softmax
9
  import logging
 
28
 
29
  # Login to Hugging Face
30
  token = os.getenv("hf_token")
 
31
  login(token)
32
 
33
  model_paths = [