Ahmed007 commited on
Commit
4f608eb
1 Parent(s): b2b346c

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -1,5 +1,6 @@
1
  from flask import Flask, request, jsonify
2
- from langchain.llms import LlamaCpp
 
3
  from langchain.callbacks.manager import CallbackManager
4
  from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
5
  from langchain.prompts import PromptTemplate
 
1
  from flask import Flask, request, jsonify
2
+ # from langchain.llms import LlamaCpp
3
+ from langchain_community.llms import LlamaCpp
4
  from langchain.callbacks.manager import CallbackManager
5
  from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
6
  from langchain.prompts import PromptTemplate