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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -9
app.py CHANGED
@@ -1,4 +1,21 @@
1
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
 
4
  import gradio as gr
@@ -9,16 +26,7 @@ import torch
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
 
1
 
2
+ import os
3
+
4
+ # Ensure Java is installed in Hugging Face Space
5
+ if not os.path.exists("/usr/bin/java"):
6
+ print("🔄 Installing Java manually...")
7
+ os.system("wget https://download.java.net/openjdk/jdk11/ri/openjdk-11+28_linux-x64_bin.tar.gz")
8
+ os.system("tar -xvzf openjdk-11+28_linux-x64_bin.tar.gz")
9
+ os.system("mv jdk-11 java")
10
+ os.environ["JAVA_HOME"] = os.path.abspath("java")
11
+ os.environ["PATH"] += os.pathsep + os.path.join(os.path.abspath("java"), "bin")
12
+
13
+ print("✅ Java is installed!")
14
+
15
+ import language_tool_python
16
+
17
+ # Initialize LanguageTool
18
+ tool = language_tool_python.LanguageTool('en-US')
19
 
20
 
21
  import gradio as gr
 
26
  from sentence_transformers import SentenceTransformer, util
27
  from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
28
 
 
 
 
 
 
29
 
 
 
 
 
30
 
31
 
32
  device = "cuda" if torch.cuda.is_available() else "cpu" # ✅ Use GPU if available