Update app.py

app.py

@@ -13,7 +13,7 @@ MAX_SEED = np.iinfo(np.int32).max
 def check_api(model_name):
     if model_name == "MAGNet":
         try :
-            client = Client("fffiloni/MAGNet", hf_token=hf_token)
+            client = Client("fffiloni/MAGNet")
             return "api ready"
         except : 
             return "api not ready yet"
@@ -244,8 +244,8 @@ def infer(image_in, chosen_model, api_status):
     if chosen_model == [] :
         raise gr.Error("Please pick a model")
 
-    #if api_status == "api not ready yet" :
-    #    raise gr.Error("This model is not ready yet, you can pick another one instead :)")
+    if api_status == "api not ready yet" :
+        raise gr.Error("This model is not ready yet, you can pick another one instead :)")
     
     gr.Info("Getting image caption with Kosmos-2...")
     user_prompt = get_caption(image_in)