cpv2280 commited on
Commit
5aa12af
·
verified ·
1 Parent(s): 1e6a001

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -9,6 +9,17 @@ import torch
9
  from sentence_transformers import SentenceTransformer, util
10
  from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
11
 
 
 
 
 
 
 
 
 
 
 
 
12
 
13
  device = "cuda" if torch.cuda.is_available() else "cpu" # ✅ Use GPU if available
14
 
 
9
  from sentence_transformers import SentenceTransformer, util
10
  from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
11
 
12
+ import os
13
+
14
+ # Ensure Java is installed in Hugging Face Space
15
+ if not os.path.exists("/usr/bin/java"):
16
+ os.system("apt-get update && apt-get install -y openjdk-11-jdk")
17
+
18
+ import language_tool_python
19
+
20
+ # Initialize LanguageTool
21
+ tool = language_tool_python.LanguageTool('en-UK')
22
+
23
 
24
  device = "cuda" if torch.cuda.is_available() else "cpu" # ✅ Use GPU if available
25