Update app.py

app.py

@@ -61,7 +61,7 @@ def download_link(content, filename, link_text):
     href = f'<a href="data:application/octet-stream;base64,{b64}" download="{filename}">{link_text}</a>'
     return href
 
-def show_predicted_caption(image, top_k=5):
+def show_predicted_caption(image, top_k=8):
     matches = predict_caption(
         image, model, text_embeddings, testing_df["caption"]
     )[:top_k]
@@ -77,7 +77,7 @@ def generate_radiology_report(prompt):
         stop=None,
         temperature=1,
     )
-    report = response.choices[1].text.strip()
+    report = response.choices[0].text.strip()
     # Remove reference string from the report
     report = re.sub(r'\(ROCO_\d+\)', '', report).strip()
     return report