Update app.py

app.py

@@ -1,11 +1,12 @@
+
+import os
+os.system("pip install deepspeed")
+
 import gradio as gr
 import re
 from magma import Magma
 from magma.image_input import ImageInput
 
-import os
-os.system("pip install deepspeed")
-
 model = Magma.from_checkpoint(
     config_path = "configs/MAGMA_v1.yml",
     checkpoint_path = "./mp_rank_00_model_states.pt",