Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 from gradio_client import Client
 from huggingface_hub import InferenceClient
 import random
+from datetime import datetime
 from models import models
 ss_client = Client("https://omnibus-html-image-current-tab.hf.space/")
 '''
@@ -16,18 +17,39 @@ models=[
     "JunRyeol/jr_model",
 ]
 '''
+
+from datetime import datetime
+
+start_time = datetime.now()
+print(start_time)
+
+while not buff.endswith('/abc #'):
+    print('waiting')
+    time_delta = datetime.now() - start_time
+    print(time_delta)
+    if time_delta.total_seconds() >= 10:
+        break
+
+
 def test_models():
     log_box=[]
     for model in models:
+        start_time = datetime.now()
         try:
             print(f'trying: {model}\n')
             client= InferenceClient(model)
             outp=client.text_generation("What is a cat")
-            log = {"Model":model,"Success":outp}
+            time_delta = datetime.now() - start_time
+            count=time_delta.total_seconds()
+            #if time_delta.total_seconds() >= 180:
+            log = {"Model":model,"Status":"Success","Output":outp, "Time":count}
             print(f'{log}\n')
             log_box.append(log)
         except Exception as e:
-            log = {"Model":model,"Error":e}
+            time_delta = datetime.now() - start_time
+            count=time_delta.total_seconds()
+
+            log = {"Model":model,"Status":"Error","Output":e,"Time":count}
             print(f'{log}\n')
             log_box.append(log)
     yield log_box