hugging2021 commited on
Commit
cbef7cc
·
verified ·
1 Parent(s): 24dd7c6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -2,10 +2,10 @@ import gradio as gr
2
  import torch
3
  from transformers import BertForMaskedLM, BertTokenizer
4
 
5
- # Modell und Tokenizer laden
6
  model_name = "bert-base-uncased"
7
- model = BertForMaskedLM.from_pretrained(model_name)
8
- tokenizer = BertTokenizer.from_pretrained(model_name)
9
 
10
  # Inferenz-Funktion definieren
11
  def inference(input_text):
@@ -44,4 +44,4 @@ iface = gr.Interface(
44
 
45
  # Interface starten
46
  if __name__ == "__main__":
47
- iface.launch(server_port=7861)
 
2
  import torch
3
  from transformers import BertForMaskedLM, BertTokenizer
4
 
5
+ # Modell und Tokenizer laden mit force_download=True
6
  model_name = "bert-base-uncased"
7
+ model = BertForMaskedLM.from_pretrained(model_name, force_download=True)
8
+ tokenizer = BertTokenizer.from_pretrained(model_name, force_download=True)
9
 
10
  # Inferenz-Funktion definieren
11
  def inference(input_text):
 
44
 
45
  # Interface starten
46
  if __name__ == "__main__":
47
+ iface.launch(server_port=7862)