Upload moondream.py
Browse files- moondream.py +1 -0
moondream.py
CHANGED
|
@@ -105,6 +105,7 @@ class Moondream(PreTrainedModel):
|
|
| 105 |
**kwargs,
|
| 106 |
)[0]
|
| 107 |
cleaned_answer = answer.strip()
|
|
|
|
| 108 |
|
| 109 |
# Use the result_queue to pass the result if it is provided
|
| 110 |
if result_queue:
|
|
|
|
| 105 |
**kwargs,
|
| 106 |
)[0]
|
| 107 |
cleaned_answer = answer.strip()
|
| 108 |
+
print("DONE")
|
| 109 |
|
| 110 |
# Use the result_queue to pass the result if it is provided
|
| 111 |
if result_queue:
|