Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -83,9 +83,4 @@ demo = gr.ChatInterface(
|
|
| 83 |
|
| 84 |
|
| 85 |
if __name__ == "__main__":
|
| 86 |
-
# Check if CUDA GPU is installed
|
| 87 |
-
if torch.cuda.is_available():
|
| 88 |
-
print("CUDA GPU is installed. Running inference on GPU.")
|
| 89 |
-
else:
|
| 90 |
-
print("CUDA GPU is not installed. Running inference on CPU.")
|
| 91 |
demo.launch()
|
|
|
|
| 83 |
|
| 84 |
|
| 85 |
if __name__ == "__main__":
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 86 |
demo.launch()
|