Antonio Toro Jaén commited on
Commit
b701d0c
·
1 Parent(s): f161c43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -1,14 +1,13 @@
1
  import gradio as gr
2
  import torch
3
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
  import re
5
  import os
6
  import csv
7
- from huggingface_hub import login
8
-
9
-
10
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
11
  import torch
 
 
 
12
 
13
  model_name = "atorojaen/DeepSeekMisongynyLyrics"
14
  bnb_config = BitsAndBytesConfig(
 
1
  import gradio as gr
2
  import torch
 
3
  import re
4
  import os
5
  import csv
 
 
 
6
  from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
7
  import torch
8
+ from huggingface_hub import login
9
+
10
+ login(os.environ["HF_TOKEN"])
11
 
12
  model_name = "atorojaen/DeepSeekMisongynyLyrics"
13
  bnb_config = BitsAndBytesConfig(