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 |
|