Update app.py

app.py

@@ -81,10 +81,10 @@ def training_computation(_dataset_dict):
     training_args = TFTrainingArguments(
         output_dir='./results',          
         num_train_epochs=2,              
-        per_device_train_batch_size=16, 
-        per_device_eval_batch_size=16,   
-        warmup_steps=500,
-        eval_steps=500, 
+        per_device_train_batch_size=128, 
+        per_device_eval_batch_size=256,   
+        warmup_steps=5,
+        eval_steps=5, 
         weight_decay=0.01               
     )
     
@@ -115,3 +115,15 @@ patent_selection = st.selectbox("Select Patent",patents['patent_number'])
 patent = patents.loc[patents['patent_number'] == patent_selection]
 st.write(patent['abstract'])
 st.write(patent['claims'])
+
+submitted = st.form_submit_button("Submit")
+
+if submitted:
+    pat_abstract = patent['abstract'].tolist()
+    #pat_score = patent['patentability_score'].tolist()
+    test_encodings = tokenizer(pat_abstract, truncation=True, padding=True)
+    test_dataset = tf.data.Dataset.from_tensor_slices((
+        dict(test_encodings)
+    ))
+    predictions = trainer.predict(test_dataset)[1]
+    st.write(predictions)