Update main.py

main.py

@@ -1,23 +1,18 @@
 from flask import Flask, request, jsonify
 from langchain_community.llms import LlamaCpp
-from langchain.callbacks.manager import CallbackManager
-from langchain.callbacks.streaming_stdout import StreamingStdOutCallbackHandler
-from langchain.prompts import PromptTemplate
-from langchain.schema.output_parser import StrOutputParser
+
 
 app = Flask(__name__)
 
 n_gpu_layers = 0
 n_batch = 1024
 
-callback_manager = CallbackManager([StreamingStdOutCallbackHandler()])
 
 llm = LlamaCpp(
     model_path="Phi-3-mini-4k-instruct-q4.gguf",  # path to GGUF file
     temperature=0.1,
     n_gpu_layers=n_gpu_layers,
     n_batch=n_batch,
-    callback_manager=callback_manager,
     verbose=True,
     n_ctx=2048
 )