Update app.py

app.py

@@ -2,12 +2,11 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import pipeline
 
-# Load AI Model for Question Extraction
-question_extractor = pipeline("text-classification", model="textattack/bert-base-uncased-MRPC")
+# Load AI Model for Question Extraction from Local Model Folder
+question_extractor = pipeline("text-classification", model="./model")
 
 app = FastAPI()
 
-# Define API Input Format
 class OCRText(BaseModel):
     text: str
 
@@ -16,7 +15,6 @@ def extract_question(data: OCRText):
     text = data.text
     lines = text.split("\n")
     
-    # Use AI Model to Identify Question Parts
     ranked_lines = sorted(lines, key=lambda line: question_extractor(line)[0]['score'], reverse=True)
     top_sentences = [line for line in ranked_lines[:3] if len(line) > 10]  # Keep Top 3 Sentences