miike-ai commited on
Commit
d7279c0
·
verified ·
1 Parent(s): 4330edf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -43,7 +43,7 @@ def extract_pdf_content(pdf_path):
43
  return " ".join(extracted_text), images
44
 
45
  # Core multimodal processing function
46
-
47
  def multimodal_chat(text_prompt, file_input=None):
48
  conversation = []
49
  images = []
@@ -81,7 +81,6 @@ def multimodal_chat(text_prompt, file_input=None):
81
 
82
  # Generate response
83
  with torch.no_grad():
84
- @spaces.GPU
85
  output = model.generate(**inputs, max_new_tokens=256)
86
 
87
  response_text = processor.decode(output[0], skip_special_tokens=True)
 
43
  return " ".join(extracted_text), images
44
 
45
  # Core multimodal processing function
46
+ @spaces.GPU
47
  def multimodal_chat(text_prompt, file_input=None):
48
  conversation = []
49
  images = []
 
81
 
82
  # Generate response
83
  with torch.no_grad():
 
84
  output = model.generate(**inputs, max_new_tokens=256)
85
 
86
  response_text = processor.decode(output[0], skip_special_tokens=True)