Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -193,11 +193,12 @@ try:
|
|
193 |
nlp = spacy.load("en_core_web_sm")
|
194 |
print("✅ Loading NLP models...")
|
195 |
|
196 |
-
# Updated summarizer: add trust_remote_code=True
|
197 |
summarizer = pipeline(
|
198 |
"summarization",
|
199 |
model="nsi319/legal-pegasus",
|
200 |
trust_remote_code=True,
|
|
|
201 |
device=0 if torch.cuda.is_available() else -1
|
202 |
)
|
203 |
embedding_model = SentenceTransformer("all-mpnet-base-v2", device=device)
|
@@ -518,7 +519,7 @@ async def analyze_legal_audio(file: UploadFile = File(...)):
|
|
518 |
temp_file_path = temp_file.name
|
519 |
print(f"Temporary file saved at: {temp_file_path}")
|
520 |
text = process_audio_to_text(temp_file_path)
|
521 |
-
if os.path.exists(
|
522 |
os.remove(temp_file_path)
|
523 |
if not text:
|
524 |
return {"status": "error", "message": "No speech could be transcribed from the audio."}
|
@@ -742,3 +743,4 @@ if __name__ == "__main__":
|
|
742 |
print("\n⚠️ Ngrok setup failed. API will only be available locally.\n")
|
743 |
run()
|
744 |
|
|
|
|
193 |
nlp = spacy.load("en_core_web_sm")
|
194 |
print("✅ Loading NLP models...")
|
195 |
|
196 |
+
# Updated summarizer: add trust_remote_code=True and use_fast=False to avoid Tiktoken conversion errors.
|
197 |
summarizer = pipeline(
|
198 |
"summarization",
|
199 |
model="nsi319/legal-pegasus",
|
200 |
trust_remote_code=True,
|
201 |
+
use_fast=False,
|
202 |
device=0 if torch.cuda.is_available() else -1
|
203 |
)
|
204 |
embedding_model = SentenceTransformer("all-mpnet-base-v2", device=device)
|
|
|
519 |
temp_file_path = temp_file.name
|
520 |
print(f"Temporary file saved at: {temp_file_path}")
|
521 |
text = process_audio_to_text(temp_file_path)
|
522 |
+
if os.path.exists(temp_audio_path):
|
523 |
os.remove(temp_file_path)
|
524 |
if not text:
|
525 |
return {"status": "error", "message": "No speech could be transcribed from the audio."}
|
|
|
743 |
print("\n⚠️ Ngrok setup failed. API will only be available locally.\n")
|
744 |
run()
|
745 |
|
746 |
+
|