Yordann commited on
Commit
af65e0a
·
verified ·
1 Parent(s): 76c40dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,9 +1,10 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
  from langdetect import detect
 
 
4
 
5
- # Load models bro my
6
-
7
  generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
8
  bg_to_en = pipeline("translation", model="Helsinki-NLP/opus-mt-bg-en")
9
  en_to_bg = pipeline("translation", model="Helsinki-NLP/opus-mt-en-bg")
 
1
  import gradio as gr
2
  from transformers import pipeline
3
  from langdetect import detect
4
+ from huggingface_hub import login
5
+ import os
6
 
7
+ login(token=os.getenv("HF_TOKEN"))
 
8
  generator = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.1")
9
  bg_to_en = pipeline("translation", model="Helsinki-NLP/opus-mt-bg-en")
10
  en_to_bg = pipeline("translation", model="Helsinki-NLP/opus-mt-en-bg")