logs

.gitignore

@@ -1,4 +1,5 @@
 logs/
+logs2/
 models
 public/*
 *.pyc

app.py

@@ -163,7 +163,7 @@ with gr.Blocks(css=css) as demo:
         with gr.Column(scale=0.85):
             txt = gr.Textbox(
                 show_label=False,
-                placeholder="Enter text and press enter. The url of the multimedia resource must contain the extension name.",
+                placeholder="Enter text and press enter. The url must contain the media type. e.g, https://example.com/example.jpg",
                 lines=1,
             ).style(container=False)
         with gr.Column(scale=0.15, min_width=0):

awesome_chat.py

@@ -1,5 +1,6 @@
 import base64
 import copy
+import datetime
 from io import BytesIO
 import io
 import os
@@ -23,6 +24,8 @@ from huggingface_hub.inference_api import InferenceApi
 from huggingface_hub.inference_api import ALL_TASKS
 from models_server import models, status
 from functools import partial
+from huggingface_hub import Repository
+import huggingface_hub
 
 parser = argparse.ArgumentParser()
 parser.add_argument("--config", type=str, default="config.yaml.dev")
@@ -37,6 +40,12 @@ config = yaml.load(open(args.config, "r"), Loader=yaml.FullLoader)
 if not os.path.exists("logs"):
     os.mkdir("logs")
 
+DATASET_REPO_URL = "https://huggingface.co/datasets/tricktreat/HuggingGPT_logs"
+LOG_HF_TOKEN = os.environ.get("LOG_HF_TOKEN")
+repo = Repository(
+    local_dir="logs", clone_from=DATASET_REPO_URL, use_auth_token=LOG_HF_TOKEN
+)
+
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.DEBUG)
 
@@ -190,13 +199,15 @@ def get_id_reason(choose_str):
     return id.strip(), reason.strip(), choose
 
 def record_case(success, **args):
+    # time format
     if success:
-        f = open("logs/log_success.jsonl", "a")
+        f = open(f"logs/log_success.jsonl", "a")
     else:
-        f = open("logs/log_fail.jsonl", "a")
+        f = open(f"logs/log_fail.jsonl", "a")
     log = args
     f.write(json.dumps(log) + "\n")
     f.close()
+    commit_url = repo.push_to_hub()
 
 def image_to_bytes(img_url):
     img_byte = io.BytesIO()

demos/demo_parse_task.json

@@ -61,5 +61,14 @@
     {
         "role": "assistant",
         "content": "[{\"task\": \"text-to-image\", \"id\": 0, \"dep\": [-1], \"args\": {\"text\": \"a picture of a cut dog\" }}, {\"task\": \"image-to-text\", \"id\": 1, \"dep\": [0], \"args\": {\"image\": \"<GENERATED>-0\" }}, {\"task\": \"text-generation\", \"id\": 2, \"dep\": [1], \"args\": {\"text\": \"<GENERATED>-1\" }}, {\"task\": \"text-to-speech\", \"id\": 3, \"dep\": [2], \"args\": {\"text\": \"<GENERATED>-2\" }}]"
+    },
+
+    {
+        "role": "user",
+        "content": "give you a picture /example.jpg, what's in it and tell me a joke about it"
+    },
+    {
+        "role": "assistant",
+        "content": "[{\"task\": \"image-to-text\", \"id\": 0, \"dep\": [-1], \"args\": {\"image\": \"/example.jpg\" }}, {\"task\": \"object-detection\", \"id\": 1, \"dep\": [-1], \"args\": {\"image\": \"/example.jpg\" }}, {\"task\": \"conversational\", \"id\": 2, \"dep\": [0], \"args\": {\"text\": \"<GENERATED>-0\" }}, {\"task\": \"text-to-speech\", \"id\": 3, \"dep\": [2], \"args\": {\"text\": \"<GENERATED>-1\" }}]"
     }
-]
+]