Darshan03 commited on
Commit
5ff9031
·
verified ·
1 Parent(s): e08fc01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -210,7 +210,8 @@ just reformulate it if needed and otherwise return it as is."""),
210
  """Structured information about cancer diagnosis and medication."""
211
  diagnosis_characteristics: List[DiagnosisCharacteristic] = field(metadata={"description": "List of primary cancers"})
212
  cancer_related_medications: List[CancerRelatedMedication] = field(metadata={"description": "List of cancer related medication given to the patient"})
213
-
 
214
  structured_llm = llm.with_structured_output(CancerInformation)
215
  try:
216
  output = structured_llm.invoke(context)
 
210
  """Structured information about cancer diagnosis and medication."""
211
  diagnosis_characteristics: List[DiagnosisCharacteristic] = field(metadata={"description": "List of primary cancers"})
212
  cancer_related_medications: List[CancerRelatedMedication] = field(metadata={"description": "List of cancer related medication given to the patient"})
213
+
214
+ llm = ChatGroq(groq_api_key=groq_api, model_name="llama-3.1-8b-instant")
215
  structured_llm = llm.with_structured_output(CancerInformation)
216
  try:
217
  output = structured_llm.invoke(context)