Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -9,6 +9,14 @@ import io
|
|
9 |
import matplotlib.pyplot as plt
|
10 |
#import pyrender
|
11 |
#import scipy
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
12 |
|
13 |
# Load the Stable Diffusion model for text-to-image generation and inpainting
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
9 |
import matplotlib.pyplot as plt
|
10 |
#import pyrender
|
11 |
#import scipy
|
12 |
+
import csv
|
13 |
+
import sys
|
14 |
+
|
15 |
+
|
16 |
+
|
17 |
+
# Increase the CSV field size limit to a large value (e.g., 10^6)
|
18 |
+
csv.field_size_limit(10**6)
|
19 |
+
|
20 |
|
21 |
# Load the Stable Diffusion model for text-to-image generation and inpainting
|
22 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|