Sharathhebbar24 commited on
Commit
4aa87a4
·
verified ·
1 Parent(s): 5f48fc7

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -0
main.py CHANGED
@@ -18,6 +18,7 @@ class UserInterface():
18
  # )
19
 
20
  models_name = (
 
21
  "HuggingFaceH4/zephyr-7b-beta",
22
  "Sharathhebbar24/chat_gpt2",
23
  "Sharathhebbar24/convo_bot_gpt_v1",
 
18
  # )
19
 
20
  models_name = (
21
+ "microsoft/DialoGPT-medium",
22
  "HuggingFaceH4/zephyr-7b-beta",
23
  "Sharathhebbar24/chat_gpt2",
24
  "Sharathhebbar24/convo_bot_gpt_v1",