Update app.py

app.py

@@ -59,9 +59,18 @@ eval = evaluator("text-classification")
 
 ### Loading models
 
+model_list=[]
 for i in range (len(models)):
+  try:
+    globals()[f"tokenizer_{i}"] = AutoTokenizer.from_pretrained(models[i])
+    globals()[f"model_{i}"] = AutoModelForSequenceClassification.from_pretrained(models[i])
+    model_list.append(models[i])
+  except:
+    print("Sorry, I can't load model "+ str(models[i]))
+
+for i in range (len(model_list)):
   globals()[f"pipe_{i}"] = pipeline("text-classification", model = models[i], tokenizer = models[i], device=-1)
-  st.text("Loaded model "+ str(models[i]))
+  st.text("Loaded pipeline "+ str(models[i]))
 
 ### Defining metrics 
 for i in range (len(metrics)):