Update llama_call.py

llama_call.py

@@ -4,7 +4,7 @@ from langchain.prompts import PromptTemplate
 from langchain_community.llms import CTransformers
 from langchain.chains import RetrievalQA
 from dotenv import load_dotenv
-from src.prompt import prompt_template
+from prompt import prompt_template
 import os
 
 load_dotenv()
@@ -48,8 +48,3 @@ qa = RetrievalQA.from_chain_type(
 def llama_call(input):
     result = qa.invoke({"query": input})
     return str(result["result"])
-
-
-if __name__ == "__main__":
-    msg = "If a previous owner of a land had allowed a neighbour or neighbour to walk or drive over his land in a shortcut and this has been going on for say a decade or so can I as the new owner stop them now from using the shortcut?"
-    print(f"response: {llama_call(msg)}")