Update app.py
Browse files
app.py
CHANGED
|
@@ -208,8 +208,9 @@ def diagnose_router(image):
|
|
| 208 |
|
| 209 |
# Generate the output from the model
|
| 210 |
output = model.generate(**inputs, max_new_tokens=300)
|
|
|
|
| 211 |
markdown_text = processor.decode(output[0])
|
| 212 |
-
|
| 213 |
# Extract JSON from the markdown text
|
| 214 |
result = extract_json_from_markdown(markdown_text)
|
| 215 |
|
|
|
|
| 208 |
|
| 209 |
# Generate the output from the model
|
| 210 |
output = model.generate(**inputs, max_new_tokens=300)
|
| 211 |
+
print(output)
|
| 212 |
markdown_text = processor.decode(output[0])
|
| 213 |
+
print(markdown_text)
|
| 214 |
# Extract JSON from the markdown text
|
| 215 |
result = extract_json_from_markdown(markdown_text)
|
| 216 |
|