Update app.py

app.py

@@ -24,8 +24,8 @@ def extract_text_from_pdf(pdf_file):
 def answer_question(question, context):
     inputs = tokenizer.encode_plus(question, context, return_tensors="pt")
     answer_start_scores, answer_end_scores = model(**inputs)
-    answer_start = torch.argmax(answer_start_scores.logits)
-    answer_end = torch.argmax(answer_end_scores.logits) + 1
+    answer_start = torch.argmax(answer_start_scores.start_logits)
+    answer_end = torch.argmax(answer_end_scores.end_logits) + 1
     answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(inputs['input_ids'][0][answer_start:answer_end]))
     return answer
 
@@ -49,7 +49,7 @@ if uploaded_file:
 
     qa_chain = ConversationalRetrievalChain(
         retriever=retriever,
-        llm=None,  # ถ้าคุณไม่ได้ใช้ HuggingFaceHub, ปรับให้เหมาะสม
+        llm=None,  # ลบ HuggingFaceHub เพราะไม่ได้ใช้งาน
         memory=memory
     )