Update src/streamlit_app.py

src/streamlit_app.py

@@ -11,7 +11,7 @@ def build_model():
     ids = tf.keras.layers.Input((MAX_LEN,), dtype=tf.int32, name="input_ids")
     att = tf.keras.layers.Input((MAX_LEN,), dtype=tf.int32, name="attention_mask")
 
-    roberta = TFRobertaModel.from_pretrained('roberta-base')
+    roberta = TFRobertaModel.from_pretrained('roberta-base', from_pt=True)
     x = roberta(ids, attention_mask=att)[0]
     x1 = tf.keras.layers.Dropout(0.1)(x) 
     start_logits = tf.keras.layers.Dense(1)(x1)