Update app.py

app.py

@@ -183,8 +183,8 @@ class VideoClassifier:
                 chain1 = LLMChain(llm=self.llm, prompt=prompt1)
                 main_class  = chain1.predict(main_categories=main_categories, transcript=transcript, captions=captions)
             except:
-                response = self.client(text=prompt1)
-                main_class = response['generated_text']
+                response = self.client.call_api(task="text-generation", inputs={"inputs": prompt1})
+                main_class = response[0]['generated_text']
                 
             print(main_class)
             print("#######################################################")
@@ -252,8 +252,10 @@ class VideoClassifier:
                 chain2 = LLMChain(llm=self.llm, prompt=prompt2)
                 answer = chain2.predict(sub_categories=sub_categories, transcript=transcript, captions=captions,main_class=main_class)
             except:
-                response = self.client(text=prompt2)
-                answer = response['generated_text']
+                # response = self.client(text=prompt2)
+                # answer = response['generated_text']
+                response = self.client.call_api(task="text-generation", inputs={"inputs": prompt2})
+                main_class = response[0]['generated_text']
             print("Preprocess Answer",answer)
 
             
@@ -338,7 +340,7 @@ class VideoClassifier:
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(description='Process some videos.')
     parser.add_argument("video_path", nargs='?', default=None, help="Path to the video file")
-    parser.add_argument("-n", "--no_of_frames", type=int, default=8, help="Number of frames for image captioning")
+    parser.add_argument("-n", "--no_of_frames", type=int, default=3, help="Number of frames for image captioning")
     parser.add_argument("--mode", choices=['interface', 'inference'], default='interface', help="Mode of operation: interface or inference")
     parser.add_argument("--model", choices=['gemini','mistral'],default='gemini',help="Model for inference")