Yermek68 commited on
Commit
9777487
·
verified ·
1 Parent(s): d226d5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,6 +1,8 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
  import huggingface_hub
 
 
4
 
5
  # ====== Настройка модели ======
6
  generator = pipeline("text-generation", model="distilgpt2")
 
1
  import gradio as gr
2
  from transformers import pipeline
3
  import huggingface_hub
4
+ import warnings
5
+ warnings.filterwarnings("ignore", category=UserWarning, module="gradio")
6
 
7
  # ====== Настройка модели ======
8
  generator = pipeline("text-generation", model="distilgpt2")