Update app.py

app.py

@@ -36,7 +36,6 @@ st.set_page_config(
 #         chunks.append(current.strip())
 #     return chunks
 
-# Save text to image
 def save_text_as_image(text, file_path):
     font = ImageFont.load_default()
     lines = text.split('\n')
@@ -128,22 +127,22 @@ if uploaded_file:
     
     prompt = PromptTemplate(input_variables=["prescription_text"], template=template)
 
-    llm = HuggingFaceEndpoint(
+    llm_model = HuggingFaceEndpoint(
         repo_id="aaditya/Llama3-OpenBioLLM-70B",
         provider="nebius",
         temperature=0.6,
         max_new_tokens=300,
-        task="text-generation"
+        task="conversational"
     )
     
-    # llm = ChatHuggingFace(
-    #     llm=llm_model,
-    #     repo_id="aaditya/Llama3-OpenBioLLM-70B",
-    #     provider="nebius",
-    #     temperature=0.6,
-    #     max_new_tokens=300,
-    #     task="conversational"
-    # )
+    llm = ChatHuggingFace(
+        llm=llm_model,
+        repo_id="aaditya/Llama3-OpenBioLLM-70B",
+        provider="nebius",
+        temperature=0.6,
+        max_new_tokens=300,
+        task="conversational"
+    )
     
     chain = LLMChain(llm=llm, prompt=prompt)