Carlosito16 commited on
Commit
971f803
1 Parent(s): c37cd1e

Update helper_function.py

Browse files
Files changed (1) hide show
  1. helper_function.py +9 -0
helper_function.py CHANGED
@@ -1,3 +1,12 @@
 
 
 
 
 
 
 
 
 
1
 
2
  def clear_input():
3
  return ("", "")
 
1
+ def get_all_models():
2
+ with open("requirements.txt") as f:
3
+ content = f.readlines()
4
+ models = []
5
+ for line in content:
6
+ if "huggingface.co" in line:
7
+ models.append(line.split("/")[4])
8
+ return models
9
+
10
 
11
  def clear_input():
12
  return ("", "")