RSK03 commited on
Commit
35df75f
·
verified ·
1 Parent(s): 2504cab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,10 +1,10 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- # Charger un modèle accessible gratuitement
5
  @st.cache_resource
6
  def load_model():
7
- return pipeline("text-generation", model="distilgpt2") # Modèle léger et libre d'accès
8
 
9
  bot = load_model()
10
 
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ # Charger un modèle en français
5
  @st.cache_resource
6
  def load_model():
7
+ return pipeline("text-generation", model="dbmdz/german-gpt2") # Modèle en français
8
 
9
  bot = load_model()
10