initial model done

README.md

@@ -1,6 +1,6 @@
 ---
 title: LLM Comparer
-emoji: 🦀
+emoji: ⚖️
 colorFrom: gray
 colorTo: pink
 sdk: gradio

app.py

@@ -1,8 +1,70 @@
 import gradio as gr
 import os
+import requests
+import openai
+
+def greet(modelpath1="",modelpath2="",modelpath3="",modelpath4="",modelpath5=""):
+    names =  [modelpath1,modelpath2,modelpath3,modelpath4,modelpath5]
+    names = [name for name in names if name != ""]
+    if names == []:
+        return "Please enter at least one model name."
+    if len(names) < 2:
+        return "Please enter at least 2 model names."
+    urls = []
+    for name in names:
+        urls.append("https://huggingface.co/" + name + "/raw/main/config.json")
+    
+    configs = []
+    index_to_ignore =[]
+    for i in range(len(urls)):
+        get_result = requests.get(urls[i])
+        if get_result.status_code == 200:
+            configs.append(get_result.json())
+        else:
+            configs.append("")
+            index_to_ignore.append(i)
+    if configs == []:
+        return "Could not find any models. Please check the model name."
+    
+    gpt_input = ""
+    gpt_input += os.environ["prompts"] +"\n\n"
+    for i in range(len(names)):
+        if i not in index_to_ignore:
+            gpt_input += "modelname: " + names[i] + "\n" + " config file:" + str(configs[i]) + "\n\n"
+
+    openai.api_key = os.environ["APIKEY"]
+
+    respose = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=[
+         {
+             "role": "system",
+             "content": gpt_input
+         },
+     ],
+    )
+    response_text = respose["choices"][0]["message"]["content"]
+    # get the first | to the last |
+    response_text = response_text[response_text.find("|")+1:response_text.rfind("|")+1]
+
+
+    return response_text
+
+text1 = gr.inputs.Textbox(placeholder="ower/modelname1", label="Input modelname like rinna/japanese-gpt-neox-3.6b",lines=1,optional=False)
+text2 = gr.inputs.Textbox(placeholder="ower/modelname2", label="model 2",lines=1,optional=False)
+text3 = gr.inputs.Textbox(placeholder="ower/modelname3", label="model 3",lines=1,optional=True)
+
+if __name__ == '__main__':
+  interFace = gr.Interface(fn=greet,
+                             inputs=[text1,text2,text3],
+                             outputs=[gr.Markdown(value="")],
+                             title="LLM Comparer⚖️",
+                             description="Plsease copy and paste the owner name / model name from the Hugging Face model hub.",
+                             theme='finlaymacklon/smooth_slate',
+                             allow_flagging=False,
+                           )
+                             
+  interFace.launch(share=False)
+    
 
-def greet(name):
-    return os.environ.get("prompts")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()

requirements.txt

@@ -0,0 +1 @@
+requests