Update frontPrompt.py
Browse files- frontPrompt.py +2 -0
frontPrompt.py
CHANGED
|
@@ -210,5 +210,7 @@ def main(image_path):
|
|
| 210 |
}
|
| 211 |
---
|
| 212 |
"""
|
|
|
|
| 213 |
response = model.chat(tokenizer, pixel_values, question, generation_config)
|
|
|
|
| 214 |
return (f'User: {question}\nAssistant: {response}')
|
|
|
|
| 210 |
}
|
| 211 |
---
|
| 212 |
"""
|
| 213 |
+
print("Before requesting model................................................................................")
|
| 214 |
response = model.chat(tokenizer, pixel_values, question, generation_config)
|
| 215 |
+
print("After requesting model................................................................................",response)
|
| 216 |
return (f'User: {question}\nAssistant: {response}')
|