Update app.py

app.py

@@ -43,7 +43,8 @@ def load_image_model(token: str):
     MODEL_NAME,
     use_auth_token=token,
     trust_remote_code=True,  # ← allow custom code in the model repo
-    device=0                 # or -1 for CPU
+    device=0,# or -1 for CPU
+    model_type = "biogpt"
     )
 
 @st.cache_resource