HannahLin271 commited on
Commit
1a048c6
·
verified ·
1 Parent(s): 5371253

Update model name

Browse files
Files changed (1) hide show
  1. app.py +5 -5
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
- "single_conversation_withoutemotion": {
33
- 'url': 'https://huggingface.co/HannahLin271/NanoGPT/resolve/main/singleConversation_withoutemotion.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_withoutemotion": {
38
- 'url': 'https://huggingface.co/HannahLin271/NanoGPT/resolve/main/wholeConversation_withoutemotion.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,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("# 🫂 Chatbot for ML Project\n### 🤗 Chat with your chatbot!")
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,