Dofla commited on
Commit
b66dc43
·
verified ·
1 Parent(s): c891b60

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
app.py CHANGED
@@ -1,10 +1,12 @@
1
 
2
- import gradio as gr
3
  from transformers import AutoTokenizer, AutoModelForQuestionAnswering
4
  import torch
 
 
 
 
5
 
6
- # Charger le tokenizer et le modèle depuis Hugging Face Spaces
7
- tokenizer = AutoTokenizer.from_pretrained("Dofla/distilbert-squad")
8
  model = AutoModelForQuestionAnswering.from_pretrained("Dofla/distilbert-squad")
9
 
10
  def question_answering(context, question):
 
1
 
 
2
  from transformers import AutoTokenizer, AutoModelForQuestionAnswering
3
  import torch
4
+ import gradio as gr
5
+
6
+ # Charger le tokenizer depuis Hugging Face Spaces
7
+ tokenizer = AutoTokenizer.from_pretrained("distilbert-squad")
8
 
9
+ # Charger le modèle depuis Hugging Face Spaces
 
10
  model = AutoModelForQuestionAnswering.from_pretrained("Dofla/distilbert-squad")
11
 
12
  def question_answering(context, question):