adding cpu

app.py

@@ -44,7 +44,7 @@ temperature = st.sidebar.slider("How randomly need you the translated sentences
 # make the process
 try:
     # recuperate checkpoints
-    checkpoints = torch.load(os.path.join(models[version][translation_type]['checkpoints'], "best_checkpoints.pth"))
+    checkpoints = torch.load(os.path.join(models[version][translation_type]['checkpoints'], "best_checkpoints.pth"), map_location=torch.device('cpu'))
 
     # recuperate the tokenizer
     tokenizer_file = models[version][translation_type]['tokenizer']
@@ -134,7 +134,7 @@ except Exception as e:
     
     st.warning("The chosen model is not available yet !", icon = "⚠️")
     
-    st.write(e)
+    # st.write(e)