fixes

.ipynb_checkpoints/app-checkpoint.py

@@ -23,7 +23,7 @@ class ImageClassificationCollator:
 
         def __call__(self, batch):
             encodings = self.feature_extractor([x[0] for x in batch], return_tensors='pt')
-            encodings['labels'] = torch.tensor([x[1] for x in batch], dtype=torch.float)
+            encodings['labels'] = torch.tensor([x[1] for x in batch], dtype=torch.long)
             return encodings 
         
 class Classifier(pl.LightningModule):
@@ -38,11 +38,16 @@ class Classifier(pl.LightningModule):
                 num_classes=model.config.num_labels
             )
 
+        # def training_step(self, batch, batch_idx):
+        #     outputs = self(**batch)
+        #     self.log(f"train_loss", outputs.loss)
+        #     return outputs.loss
         def training_step(self, batch, batch_idx):
             outputs = self(**batch)
-            self.log(f"train_loss", outputs.loss)
-            return outputs.loss
-
+            loss = outputs.loss.float()  # Convert to float
+            self.log(f"train_loss", loss)
+            return loss
+        
         def validation_step(self, batch, batch_idx):
             outputs = self(**batch)
             self.log(f"val_loss", outputs.loss)
@@ -138,8 +143,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
             frameNr = frameNr+10
 
         img=cv2.imread(class_d+'/frame_0.jpg')
-        a=str(transformers.__version__)
-        return img, a, class_d
+        return img, user_d, class_d
 demo = gr.Interface(video_identity, 
                     inputs=[gr.Video(source='upload'),
                             gr.Text(),

app.py

@@ -23,7 +23,7 @@ class ImageClassificationCollator:
 
         def __call__(self, batch):
             encodings = self.feature_extractor([x[0] for x in batch], return_tensors='pt')
-            encodings['labels'] = torch.tensor([x[1] for x in batch], dtype=torch.float)
+            encodings['labels'] = torch.tensor([x[1] for x in batch], dtype=torch.long)
             return encodings 
         
 class Classifier(pl.LightningModule):
@@ -38,11 +38,16 @@ class Classifier(pl.LightningModule):
                 num_classes=model.config.num_labels
             )
 
+        # def training_step(self, batch, batch_idx):
+        #     outputs = self(**batch)
+        #     self.log(f"train_loss", outputs.loss)
+        #     return outputs.loss
         def training_step(self, batch, batch_idx):
             outputs = self(**batch)
-            self.log(f"train_loss", outputs.loss)
-            return outputs.loss
-
+            loss = outputs.loss.float()  # Convert to float
+            self.log(f"train_loss", loss)
+            return loss
+        
         def validation_step(self, batch, batch_idx):
             outputs = self(**batch)
             self.log(f"val_loss", outputs.loss)
@@ -138,8 +143,7 @@ def video_identity(video,user_name,class_name,trainortest,ready):
             frameNr = frameNr+10
 
         img=cv2.imread(class_d+'/frame_0.jpg')
-        a=str(transformers.__version__)
-        return img, a, class_d
+        return img, user_d, class_d
 demo = gr.Interface(video_identity, 
                     inputs=[gr.Video(source='upload'),
                             gr.Text(),