Hugging Face's logo

Spaces:
Spestly
/
AtlasUI
Sleeping

App Files Community
1
Spestly commited on
Commit
b2efec6
·
verified ·
1 Parent(s): be78969

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
- from transformers import AutoTokenizer, AutoModelForCausalLM, login
 
3
  import torch
4
 
5
  HF_TOKEN = os.getenv("HF_TOKEN")
 
1
  import gradio as gr
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM
3
+ from huggingface_hub import login
4
  import torch
5
 
6
  HF_TOKEN = os.getenv("HF_TOKEN")