nafisneehal commited on
Commit
d913f1a
·
verified ·
1 Parent(s): f177164

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -4,6 +4,14 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, GenerationConfig,
4
  import spaces
5
  import os
6
  import json
 
 
 
 
 
 
 
 
7
 
8
  # File to store model links
9
  MODEL_FILE = "model_links.txt"
 
4
  import spaces
5
  import os
6
  import json
7
+ from huggingface_hub import login
8
+
9
+ # Hugging Face authentication
10
+ HF_TOKEN = secrets.HF_TOKEN
11
+ try:
12
+ login(token=HF_TOKEN)
13
+ except Exception as e:
14
+ print(f"Error logging in to Hugging Face: {str(e)}")
15
 
16
  # File to store model links
17
  MODEL_FILE = "model_links.txt"