Update app.py

app.py

@@ -54,7 +54,6 @@ st.markdown("### Please wait for the dataset and models to load (this can take s
 @st.cache 
 def loaddset(d, d_split):
   data = datasets.load_dataset(d, split=d_split)
-  st.text("Loaded the "+ str(d_split)+ " split of dataset "+ str(d))
   return(data)
 
 data = loaddset(dset,dset_split)
@@ -72,17 +71,18 @@ def load_models(mod_names):
       globals()[f"model_{i}"] = AutoModelForSequenceClassification.from_pretrained(mod_names[i])
       model_list.append(mod_names[i])
     except:
-      print("Sorry, I can't load model "+ str(mod_names[i]))
-  return("Loaded "+ str(len(model_list))+ " models")
+      continue
+  return(model_list)
 
 load_models(models)
 
 @st.cache 
 def load_pipes(mod_list):
+  pipe_list=[]
   for i in range (len(mod_list)):
     globals()[f"pipe_{i}"] = pipeline("text-classification", model = models[i], tokenizer = models[i], device=-1)
-    st.text("Loaded pipeline "+ str(models[i]))
-  return("Loaded "+ str(len(mod_list))+ " models") 
+     
+  return(pipe_list) 
 
 load_pipes(model_list)