Update app.py

app.py

@@ -22,6 +22,19 @@ def generate(
         temperature = 1e-2
     top_p = float(top_p)
 
+    template1 = '''Given below are the different type of main video classes  
+        {main_categories}
+        You are a text classifier that catergorises the transcript and captions into one main class whose context match with one main class and only generate main class name no need of sub classe or explanation.
+        Give more importance to Transcript while classifying .
+        Transcript: {transcript}
+        Captions: {captions}
+        Return only the answer chosen from list and nothing else
+        Main-class =>  '''
+    main_categories="x"
+    transcript="y"
+    captions="z"
+    prompt1 = PromptTemplate(template=template1, input_variables=['main_categories', 'transcript', 'captions'])
+    
     generate_kwargs = dict(
         temperature=temperature,
         max_new_tokens=max_new_tokens,
@@ -33,7 +46,7 @@ def generate(
 
     formatted_prompt = format_prompt(prompt, history)
 
-    stream = client.text_generation(formatted_prompt, **generate_kwargs, stream=True, details=True, return_full_text=False)
+    stream = client.text_generation(template1, **generate_kwargs, stream=True, details=True, return_full_text=False)
     output = ""
 
     for response in stream: