fixes

.ipynb_checkpoints/app-checkpoint.py

@@ -38,16 +38,11 @@ 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)
-            loss = outputs.loss.float()  # Convert to float
-            self.log(f"train_loss", loss)
-            return loss
-        
+            self.log(f"train_loss", outputs.loss)
+            return outputs.loss
+
         def validation_step(self, batch, batch_idx):
             outputs = self(**batch)
             self.log(f"val_loss", outputs.loss)
@@ -141,9 +136,9 @@ def video_identity(video,user_name,class_name,trainortest,ready):
                 break
 
             frameNr = frameNr+10
-
+        a=pl.__version__
         img=cv2.imread(class_d+'/frame_0.jpg')
-        return img, user_d, class_d
+        return img, a, class_d
 demo = gr.Interface(video_identity, 
                     inputs=[gr.Video(source='upload'),
                             gr.Text(),

app.py

@@ -38,16 +38,11 @@ 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)
-            loss = outputs.loss.float()  # Convert to float
-            self.log(f"train_loss", loss)
-            return loss
-        
+            self.log(f"train_loss", outputs.loss)
+            return outputs.loss
+
         def validation_step(self, batch, batch_idx):
             outputs = self(**batch)
             self.log(f"val_loss", outputs.loss)
@@ -141,9 +136,9 @@ def video_identity(video,user_name,class_name,trainortest,ready):
                 break
 
             frameNr = frameNr+10
-
+        a=pl.__version__
         img=cv2.imread(class_d+'/frame_0.jpg')
-        return img, user_d, class_d
+        return img, a, class_d
 demo = gr.Interface(video_identity, 
                     inputs=[gr.Video(source='upload'),
                             gr.Text(),