Skip to content

Commit

Permalink
FIX: add missing pdf_id
Browse files Browse the repository at this point in the history
  • Loading branch information
CodingWithTim committed Feb 19, 2025
1 parent 3528919 commit bac54f0
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions fastchat/serve/gradio_web_server.py
Original file line number Diff line number Diff line change
Expand Up @@ -121,6 +121,9 @@ def __init__(self, model_name, is_vision=False, pdf_id=None):
self.is_vision = is_vision
self.ans_models = []
self.router_outputs = []
self.pdf_id = (
pdf_id # NOTE(Tim): Version 1 PDFChat Architecture, could be revised later.
)

# NOTE(chris): This could be sort of a hack since it assumes the user only uploads one image. If they can upload multiple, we should store a list of image hashes.
self.has_csam_image = False
Expand Down

0 comments on commit bac54f0

Please sign in to comment.