Update app.py
Browse files
app.py
CHANGED
@@ -51,7 +51,7 @@ headers = {"Authorization": f"Bearer {API_TOKEN}"}
|
|
51 |
timeout = 100
|
52 |
|
53 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
54 |
-
model_repo_id = "
|
55 |
|
56 |
if torch.cuda.is_available():
|
57 |
torch_dtype = torch.float16
|
|
|
51 |
timeout = 100
|
52 |
|
53 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
54 |
+
model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
|
55 |
|
56 |
if torch.cuda.is_available():
|
57 |
torch_dtype = torch.float16
|