Update app.py

app.py

@@ -2,8 +2,8 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import pipeline
 
-# Load AI Model for Question Extraction from Local Model Folder
-question_extractor = pipeline("text-classification", model="./model")
+# ✅ Use Hugging Face Hub Instead of Local Model
+question_extractor = pipeline("text-classification", model="textattack/bert-base-uncased-MRPC")
 
 app = FastAPI()
 
@@ -16,8 +16,9 @@ def extract_question(data: OCRText):
     lines = text.split("\n")
     
     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
+    top_sentences = [line for line in ranked_lines[:3] if len(line) > 10]  
     
     question_text = " ".join(top_sentences)
     
     return {"extracted_question": question_text}
+