Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
|
@@ -11,7 +11,7 @@ logging.basicConfig(level=logging.INFO)
|
|
| 11 |
logger = logging.getLogger(__name__)
|
| 12 |
|
| 13 |
app = FastAPI()
|
| 14 |
-
MODEL_NAME = 'krishna_choudhary/
|
| 15 |
|
| 16 |
def get_llm():
|
| 17 |
callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
|
|
|
|
| 11 |
logger = logging.getLogger(__name__)
|
| 12 |
|
| 13 |
app = FastAPI()
|
| 14 |
+
MODEL_NAME = 'krishna_choudhary/lightweight_chatbot'
|
| 15 |
|
| 16 |
def get_llm():
|
| 17 |
callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
|