Spaces:
Sleeping
Sleeping
OmkarThawakar
commited on
Commit
·
c2d589f
1
Parent(s):
8029250
change device setting
Browse files
app.py
CHANGED
|
@@ -169,7 +169,7 @@ class Chat:
|
|
| 169 |
|
| 170 |
|
| 171 |
|
| 172 |
-
chat = Chat(model,transform,df,tar_img_feats)
|
| 173 |
print("Chat Initialized !")
|
| 174 |
|
| 175 |
|
|
|
|
| 169 |
|
| 170 |
|
| 171 |
|
| 172 |
+
chat = Chat(model,transform,df,tar_img_feats, device)
|
| 173 |
print("Chat Initialized !")
|
| 174 |
|
| 175 |
|