Spaces:

ragavsachdeva
/

Magiv2-Demo

Runtime error

ragavsachdeva commited on Aug 15, 2024

Commit

9d0818e

verified ·

1 Parent(s): f6db6a4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,7 @@ import numpy as np
 from transformers import AutoModel
 import torch
 import spaces
 # Load the model
 model = AutoModel.from_pretrained("ragavsachdeva/magiv2", trust_remote_code=True).cuda().eval()

 from transformers import AutoModel
 import torch
 import spaces
+import os
 # Load the model
 model = AutoModel.from_pretrained("ragavsachdeva/magiv2", trust_remote_code=True).cuda().eval()