Woziii commited on
Commit
fb8a6d9
·
verified ·
1 Parent(s): ac49c2f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -4
app.py CHANGED
@@ -1,18 +1,32 @@
 
 
 
 
 
1
  import json
2
- import re
3
  import logging
4
- from transformers import AutoModelForCausalLM, AutoTokenizer
5
- import gradio as gr
 
 
 
6
 
7
- # Configuration du logger
8
  logging.basicConfig(
9
  level=logging.INFO,
10
  format="%(asctime)s - %(levelname)s - %(message)s",
11
  handlers=[
 
12
  logging.StreamHandler()
13
  ]
14
  )
15
 
 
 
 
 
 
 
16
  # Variables globales simulées
17
  project_state = {
18
  "AgentManager": {"structured_summary": None},
 
1
+ import gradio as gr
2
+ import torch
3
+ from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer, pipeline
4
+ from datetime import datetime
5
+ import os
6
  import json
 
7
  import logging
8
+ from huggingface_hub import login
9
+ import requests
10
+ from bs4 import BeautifulSoup
11
+ from concurrent.futures import ThreadPoolExecutor
12
+ import re
13
 
14
+ # --- Configuration du logger ---
15
  logging.basicConfig(
16
  level=logging.INFO,
17
  format="%(asctime)s - %(levelname)s - %(message)s",
18
  handlers=[
19
+ logging.FileHandler("project.log"),
20
  logging.StreamHandler()
21
  ]
22
  )
23
 
24
+ # --- Authentification Hugging Face ---
25
+ # Assurez-vous que la variable d'environnement HF_TOKEN est définie avec votre token Hugging Face
26
+ # Sinon, vous pouvez la définir directement ici
27
+ # os.environ["HF_TOKEN"] = "votre_token_huggingface"
28
+
29
+ login(token=os.environ["HF_TOKEN"])
30
  # Variables globales simulées
31
  project_state = {
32
  "AgentManager": {"structured_summary": None},