Update app.py

app.py

@@ -62,7 +62,12 @@ def create_transformer_model(input_shape):
     x = transformer_encoder(inputs, head_size=256, num_heads=4, ff_dim=4, dropout=0.1)
     x = GlobalAveragePooling1D()(x)
     outputs = Dense(1)(x)
-    return Model(inputs, outputs)
+    model = Model(inputs, outputs)
+    
+    # Compile the model
+    model.compile(optimizer='adam', loss='mse')
+    
+    return model
 
 def transformer_encoder(inputs, head_size, num_heads, ff_dim, dropout=0):
     x = MultiHeadAttention(key_dim=head_size, num_heads=num_heads, dropout=dropout)(inputs, inputs)