Spaces:
Running
Running
HannahLin271
commited on
Update model name
Browse files
app.py
CHANGED
@@ -29,13 +29,13 @@ model_info = {
|
|
29 |
'description': "Trained on Facebook Emotion Dialogues dataset, retaining emotion annotations for each conversation, using a default batch size of 64.",
|
30 |
'logo': '🍺'
|
31 |
},
|
32 |
-
"
|
33 |
-
'url': 'https://huggingface.co/HannahLin271/NanoGPT/resolve/main/
|
34 |
'description': "Trained on Facebook Emotion Dialogues dataset, excluding emotion annotations for simpler conversations, using a default batch size of 64.",
|
35 |
'logo': '🍷'
|
36 |
},
|
37 |
-
"
|
38 |
-
'url': 'https://huggingface.co/HannahLin271/NanoGPT/resolve/main/
|
39 |
'description': "Trained on entire conversations from the Facebook Emotion Dialogues dataset, excluding tags other than <bot> and <human>,, using a default batch size of 64",
|
40 |
'logo': '🍵'
|
41 |
}
|
@@ -102,7 +102,7 @@ default_model_info = f"## 🍭Model Information\n<br>Model_name: single_conversa
|
|
102 |
app = gr.Blocks()
|
103 |
full_output = " "
|
104 |
with app:
|
105 |
-
gr.Markdown("# 🫂
|
106 |
# Model Parameters interface
|
107 |
inp = gr.Dropdown(
|
108 |
choices=model_choices,
|
|
|
29 |
'description': "Trained on Facebook Emotion Dialogues dataset, retaining emotion annotations for each conversation, using a default batch size of 64.",
|
30 |
'logo': '🍺'
|
31 |
},
|
32 |
+
"single_conversation": {
|
33 |
+
'url': 'https://huggingface.co/HannahLin271/NanoGPT/resolve/main/singleConversation.pt',
|
34 |
'description': "Trained on Facebook Emotion Dialogues dataset, excluding emotion annotations for simpler conversations, using a default batch size of 64.",
|
35 |
'logo': '🍷'
|
36 |
},
|
37 |
+
"whole_conversation": {
|
38 |
+
'url': 'https://huggingface.co/HannahLin271/NanoGPT/resolve/main/wholeConversation.pt',
|
39 |
'description': "Trained on entire conversations from the Facebook Emotion Dialogues dataset, excluding tags other than <bot> and <human>,, using a default batch size of 64",
|
40 |
'logo': '🍵'
|
41 |
}
|
|
|
102 |
app = gr.Blocks()
|
103 |
full_output = " "
|
104 |
with app:
|
105 |
+
gr.Markdown("# 🫂 Empathetic Dialogue System \n🤗For the chatbot, we trained a small language model from scratch in our local machine. You can find the detailed information about our project in the Github repository [here](https://github.com/CallMeL/Project-ML).")
|
106 |
# Model Parameters interface
|
107 |
inp = gr.Dropdown(
|
108 |
choices=model_choices,
|