Update app.py
Browse files
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=
|
|
|
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)
|