Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -19,6 +19,7 @@ from torch import nn
|
|
| 19 |
import requests
|
| 20 |
from numpy.lib.type_check import imag
|
| 21 |
import random
|
|
|
|
| 22 |
|
| 23 |
import csv
|
| 24 |
import torch
|
|
@@ -169,21 +170,22 @@ def inference(image_url, image, min_score):
|
|
| 169 |
|
| 170 |
|
| 171 |
# Write the measurements to a CSV file
|
| 172 |
-
|
|
|
|
| 173 |
writer = csv.writer(file)
|
| 174 |
writer.writerow(['ID', 'Measurement', 'X_Min', 'X_Max', 'Y_Min', 'Y_Max'])
|
| 175 |
for id, data in measurements.items():
|
| 176 |
writer.writerow([id, data['measurement'], data['x_min'], data['x_max'], data['y_min'], data['y_max']])
|
| 177 |
# Convert the CSV content to a bytes object
|
| 178 |
-
|
| 179 |
-
|
| 180 |
-
|
| 181 |
-
|
| 182 |
-
|
| 183 |
-
|
| 184 |
-
|
| 185 |
-
|
| 186 |
-
|
| 187 |
|
| 188 |
return upload_result["url"]
|
| 189 |
|
|
|
|
| 19 |
import requests
|
| 20 |
from numpy.lib.type_check import imag
|
| 21 |
import random
|
| 22 |
+
import time
|
| 23 |
|
| 24 |
import csv
|
| 25 |
import torch
|
|
|
|
| 170 |
|
| 171 |
|
| 172 |
# Write the measurements to a CSV file
|
| 173 |
+
filename=str(time.time())+'dmeasurements.csv'
|
| 174 |
+
with open(filename, mode='w') as file:
|
| 175 |
writer = csv.writer(file)
|
| 176 |
writer.writerow(['ID', 'Measurement', 'X_Min', 'X_Max', 'Y_Min', 'Y_Max'])
|
| 177 |
for id, data in measurements.items():
|
| 178 |
writer.writerow([id, data['measurement'], data['x_min'], data['x_max'], data['y_min'], data['y_max']])
|
| 179 |
# Convert the CSV content to a bytes object
|
| 180 |
+
|
| 181 |
+
# csv_bytes = io.StringIO( file).read().encode("utf-8")
|
| 182 |
+
|
| 183 |
+
# Upload the file to Cloudinary
|
| 184 |
+
upload_result = cloudinary.uploader.upload(
|
| 185 |
+
filename,
|
| 186 |
+
folder = "csv_files"
|
| 187 |
+
)
|
| 188 |
+
# return file
|
| 189 |
|
| 190 |
return upload_result["url"]
|
| 191 |
|