Spaces:
Build error
Build error
Update app.py
Browse files
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)
|