Spaces:
Runtime error
Runtime error
Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,10 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from persona_chatbot import PersonaChatbot
|
| 2 |
+
|
| 3 |
+
MODEL_PATH = "path/to/your/model.gguf"
|
| 4 |
+
DATASET_NAME = "K00B404/personas"
|
| 5 |
+
|
| 6 |
+
bot = PersonaChatbot(MODEL_PATH, DATASET_NAME)
|
| 7 |
+
bot.list_personas()
|
| 8 |
+
choice = int(input("Select a persona by number: "))
|
| 9 |
+
bot.select_persona(choice)
|
| 10 |
+
bot.chat()
|