Update app.py
Browse files
app.py
CHANGED
|
@@ -94,9 +94,6 @@ class ImageClassifier:
|
|
| 94 |
logit, predicted = torch.max(output.data, 1)
|
| 95 |
return self.labels[predicted[0].item()], logit[0].item()
|
| 96 |
|
| 97 |
-
|
| 98 |
-
|
| 99 |
-
@spaces.GPU()
|
| 100 |
def pipeline_to_setup_with_gpu():
|
| 101 |
run_commands()
|
| 102 |
sys.path.append(join(dirname(abspath(__file__)),'GroundingDINO'))
|
|
|
|
| 94 |
logit, predicted = torch.max(output.data, 1)
|
| 95 |
return self.labels[predicted[0].item()], logit[0].item()
|
| 96 |
|
|
|
|
|
|
|
|
|
|
| 97 |
def pipeline_to_setup_with_gpu():
|
| 98 |
run_commands()
|
| 99 |
sys.path.append(join(dirname(abspath(__file__)),'GroundingDINO'))
|