Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,23 @@ import os
|
|
2 |
import gradio as gr
|
3 |
import spaces
|
4 |
import torch
|
|
|
5 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
6 |
|
7 |
model_id = "futurehouse/ether0"
|
8 |
|
|
|
2 |
import gradio as gr
|
3 |
import spaces
|
4 |
import torch
|
5 |
+
import logging
|
6 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
7 |
+
from transformers.utils import logging as hf_logging
|
8 |
+
|
9 |
+
logging.basicConfig(
|
10 |
+
filename="/tmp/app.log",
|
11 |
+
level=logging.DEBUG,
|
12 |
+
format="%(asctime)s %(levelname)s: %(message)s"
|
13 |
+
)
|
14 |
+
|
15 |
+
logging.info("Starting app.py logging")
|
16 |
+
hf_logging.set_verbosity_debug()
|
17 |
+
hf_logging.set_verbosity_info()
|
18 |
+
hf_logging.enable_default_handler()
|
19 |
+
hf_logging.enable_explicit_format()
|
20 |
+
hf_logging.add_handler(logging.FileHandler("/tmp/transformers.log"))
|
21 |
+
|
22 |
|
23 |
model_id = "futurehouse/ether0"
|
24 |
|