Update app.py

app.py

@@ -3,8 +3,8 @@ from trulens_eval.schema import Select
 from trulens_eval.tru import Tru
 from trulens_eval.feedback import Feedback
 from trulens_eval.feedback import OpenAI as Feedback_OpenAI
-from langchain import HuggingFacePipeline
 from langchain.llms import HuggingFacePipeline
+from langchain.prompts import PromptTemplate
 from langchain.chains import ConversationChain
 from langchain.chains.conversation.memory import ConversationBufferWindowMemory
 import os
@@ -23,9 +23,9 @@ tru = Tru()
 
 # Initialize the HuggingFacePipeline for local LLM
 local_llm = HuggingFacePipeline.from_model_id(
-    model_id="chavinlo/alpaca-native",
+    model_id="Tonic/stablemed",
     task="text-generation",
-    model_kwargs={"temperature": 0.6, "top_p": 0.95, "max_length": 256}
+    model_kwargs={"temperature": 0.2, "top_p": 0.95, "max_length": 256}
 )
 
 # Set the window memory to go back 4 turns
@@ -39,7 +39,7 @@ conversation = ConversationChain(
 )
 
 # Update the conversation prompt template to prime it as a gardening expert
-conversation.prompt.template = '''The following is a friendly conversation between a human and an AI gardening expert. The AI is an expert on gardening and gives recommendations specific to location and conditions. If the AI does not know the answer to a question, it truthfully says it does not know.
+conversation.prompt.template = '''The following is a consult between a clinical consultant and a public health and medical expert. The AI is an expert on medicine and public health and gives recommendations specific to location and conditions. If the AI does not know the answer to a question, it truthfully says it does not know.
 
 Current conversation:
 {history}
@@ -47,7 +47,7 @@ Human: {input}
 AI:'''
 
 # Wrap the conversation with TruChain to instrument it
-tc_conversation = tru.Chain(conversation, app_id='GardeningAIwithMemory_v1', feedbacks=[qa_relevance])
+tc_conversation = tru.Chain(conversation, app_id='Trulens-StableMed', feedbacks=[qa_relevance])
 
 # Initialize Gradio Client
 client = Client("https://tonic-stablemed-chat.hf.space/")