Update app.py

app.py

@@ -10,7 +10,7 @@ import easyocr
 # from langchain_community.document_loaders import UnstructuredImageLoader
 from langchain.prompts import PromptTemplate
 from langchain.chains import LLMChain
-from langchain_huggingface import HuggingFaceEndpoint
+from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
 
 # Set Hugging Face API keys
 os.environ["HUGGINGFACEHUB_API_KEY"] = os.getenv("HF")
@@ -83,13 +83,23 @@ if uploaded_file:
     """
     prompt = PromptTemplate(input_variables=["prescription_text"], template=template)
 
-    model = HuggingFaceEndpoint(
+    llm_model = HuggingFaceEndpoint(
         repo_id="mistralai/Mistral-7B-Instruct-v0.3",
         provider="novita",
         temperature=0.6,
         max_new_tokens=300,
         task="conversational"
     )
+
+    model = ChatHuggingFace(
+        llm=llm_model,
+        repo_id="mistralai/Mistral-7B-Instruct-v0.3",
+        provider="novita",
+        temperature=0.6,
+        max_new_tokens=300,
+        task="conversational"
+    )
+    
     chain = LLMChain(llm=model, prompt=prompt)
 
    
@@ -117,7 +127,6 @@ if uploaded_file:
     os.remove(dilated_path)
 
 else:
-    st.image("https://cdn.dribbble.com/users/285475/screenshots/14711920/media/bd46dc2873f7099e4ef9fd53e6f7f1df.png", width=600)
     st.markdown("<center><i>Upload a prescription image to begin analysis.</i></center>", unsafe_allow_html=True)