Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,21 @@
|
|
2 |
##Trying List And Create SubFolder
|
3 |
|
4 |
import gradio as gr
|
5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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()
|