Removing of gpt2

app.py

@@ -16,26 +16,16 @@ input_text = st.text_input("Testo:", value="Lorem ipsum dolor sit amet, [MASK] a
 #modelname = "./models/latin_bert/"
 #modelname = "LuisAVasquez/simple-latin-bert-uncased"
 modelname = "./models/bert-base-latin-uncased"
-# Model based on GPT 2
-modelname_gpt = "itserr/scratch_2-nodes_tokenizer_latbert-original_packing_fcocchi"
+
 
 tokenizer = AutoTokenizer.from_pretrained(modelname)
 model = AutoModelForMaskedLM.from_pretrained(modelname)
 fill_mask = pipeline("fill-mask", model=model, tokenizer=tokenizer)
 
-#Use GPT 2 to compare with BERT
-tokenizer_gpt = AutoTokenizer.from_pretrained(modelname_gpt)
-model_gpt = AutoModelForMaskedLM.from_pretrained(modelname_gpt)
-fill_mask_gpt = pipeline("fill-mask", model=model_gpt, tokenizer=tokenizer_gpt)
-
 if input_text:
     predictions = fill_mask(input_text)
     st.subheader("Risultati delle previsioni con Simple Latin Bert:")
     for pred in predictions:
         st.write(f"**Parola**: {pred['token_str']}, **Probabilità**: {pred['score']:.4f}, **Sequence**: {pred['sequence']}")
-    predictions_gpt = fill_mask_gpt(input_text)
-    st.subheader("Risultati delle previsioni con Simple Latin GPT2:")
-    for pred_gpt in predictions_gpt:
-        st.write(f"**Parola**: {pred_gpt['token_str']}, **Probabilità**: {pred_gpt['score']:.4f}, **Sequence**: {pred_gpt['sequence']}")