Update main.py

main.py

@@ -22,7 +22,7 @@ def preprocess_text(question: str):
 def classify_text(question: str):
     prompt_template = PromptTemplate(template="Answer the following question and classify it: {question}",
                                      input_variables=["question"])
-    format_prompt = prompt_template.format(question=input_text)
+    format_prompt = prompt_template.format(question=question)
     encoded_input = tokenizer(format_prompt, return_tensors='pt')
     # Run the model
     output = model.generate(**encoded_input)  # Use generate method for text generation