Update app.py

app.py

@@ -22,7 +22,10 @@ def predict(vector):
     
     # 生成答案
     tokenizer = RagTokenizer.from_pretrained("facebook/rag-sequence-nq")
-    model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-nq")
+    model = RagSequenceForGeneration.from_pretrained(
+        "facebook/rag-sequence-nq",
+        torch_dtype=torch.float16
+    )
     inputs = tokenizer.prepare_seq2seq_batch(
         [vector], 
         return_tensors="pt"