Update app.py

app.py

@@ -10,9 +10,12 @@ def model_explorer(search,task,limit=100):
         search=None
     out_box=[]
     name_box=[]
-    filt = ModelFilter(task="text-generation")
-    
-    this = api.list_models(search=search,limit=limit,filter=filt,cardData=True)
+    if task:
+        filt = ModelFilter(task=task)
+        this = api.list_models(search=search,limit=limit,filter=filt,cardData=True)
+    else:
+        this = api.list_models(search=search,limit=limit,cardData=True)
+        
     cnt=0
     for i,mod in enumerate(this):
         cnt+=1