Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,7 +6,7 @@ import os
|
|
6 |
|
7 |
# Model and tokenizer details
|
8 |
model_repo = "elapt1c/ElapticAI-1a"
|
9 |
-
model_filename = "
|
10 |
tokenizer_name = "microsoft/DialoGPT-medium"
|
11 |
|
12 |
# Device configuration
|
|
|
6 |
|
7 |
# Model and tokenizer details
|
8 |
model_repo = "elapt1c/ElapticAI-1a"
|
9 |
+
model_filename = "model.pth" # Assuming the model is saved as pytorch_model.bin, adjust if needed. Check the HF repo.
|
10 |
tokenizer_name = "microsoft/DialoGPT-medium"
|
11 |
|
12 |
# Device configuration
|