MisterAI commited on
Commit
08ae3bc
·
verified ·
1 Parent(s): c5f5c24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -1
app.py CHANGED
@@ -2,7 +2,21 @@
2
  ##Trying List And Create SubFolder
3
 
4
  import gradio as gr
5
- from huggingface_hub import HfApi
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
 
7
  def list_repo_content(repo_path):
8
  api = HfApi()
 
2
  ##Trying List And Create SubFolder
3
 
4
  import gradio as gr
5
+ import os
6
+ from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingArguments, DataCollatorForLanguageModeling
7
+ from datasets import load_dataset, Dataset
8
+ from huggingface_hub import HfApi, HfFolder
9
+ import requests
10
+ from io import BytesIO
11
+
12
+ # Récupérer token depuis les variables d'environnement
13
+ hf_token = os.getenv("MisterAI_bigscience_bloom_560m")
14
+
15
+ # Configurer le token pour l'utilisation avec Hugging Face
16
+ if hf_token:
17
+ HfFolder.save_token(hf_token)
18
+ else:
19
+ raise ValueError("Le token Hugging Face n'est pas configuré. Assurez-vous qu'il est défini dans les variables d'environnement.")
20
 
21
  def list_repo_content(repo_path):
22
  api = HfApi()