Spaces:
Runtime error
Runtime error
Commit
•
3c4e180
1
Parent(s):
916af8f
imported os
Browse files
app.py
CHANGED
|
@@ -7,6 +7,7 @@ import time
|
|
| 7 |
import gradio as gr
|
| 8 |
import numpy as np
|
| 9 |
from pathlib import Path
|
|
|
|
| 10 |
|
| 11 |
class_names = ['pizza','steak','sushi']
|
| 12 |
device = 'cuda' if torch.cuda.is_available else 'cpu'
|
|
|
|
| 7 |
import gradio as gr
|
| 8 |
import numpy as np
|
| 9 |
from pathlib import Path
|
| 10 |
+
import os
|
| 11 |
|
| 12 |
class_names = ['pizza','steak','sushi']
|
| 13 |
device = 'cuda' if torch.cuda.is_available else 'cpu'
|