Update app.py

app.py

@@ -5,7 +5,7 @@ from mingru_lm import MinGRU_LM
 
 model = MinGRU_LM(dim=512, num_tokens=256, num_layers=6)
 pt_model = "best_model.pt"
-checkpoint = torch.load(pt_model,map_location=torch.device('cpu'))
+checkpoint = torch.load(pt_model,map_location=torch.device('cpu'),weights_only=True)
 model.load_state_dict(checkpoint['model_state_dict'])
 
 # Move model to GPU if available
@@ -57,4 +57,4 @@ iface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    iface.launch()
+    iface.launch(show_api=False, server_name="0.0.0.0")