omanaaja commited on
Commit
aadbe2e
·
1 Parent(s): a2fc960

penyesuaian app.py dan update dependencies requirements.txt

Browse files
Files changed (2) hide show
  1. app.py +8 -4
  2. requirements.txt +3 -0
app.py CHANGED
@@ -10,10 +10,14 @@ from io import BytesIO
10
  from PIL import Image
11
  import uvicorn
12
  from fastapi.middleware.cors import CORSMiddleware
 
 
13
 
14
-
15
- # Load model dan label
16
  model = load_model_from_files("model.json", "my_model.h5")
 
 
17
 
18
  labels = [
19
  "Benteng Vredeburg", "Candi Borobudur", "Candi Prambanan", "Gedung Agung Istana Kepresidenan",
@@ -89,13 +93,13 @@ gradio_app = gr.Interface(
89
  gr.Textbox(label="Deskripsi Lengkap", lines=20, max_lines=50),
90
  gr.HTML(label="Link Lokasi"),
91
  ],
92
- allow_flagging="never",
93
  title="Klasifikasi Gambar",
94
  description="Upload gambar, sistem akan mengklasifikasikan dan memberikan deskripsi mengenai gambar tersebut."
95
  )
96
 
97
  # Mount Gradio ke FastAPI
98
- app = gr.mount_gradio_app(app, gradio_app, path="")
99
 
100
  # Jalankan app
101
  if __name__ == "__main__":
 
10
  from PIL import Image
11
  import uvicorn
12
  from fastapi.middleware.cors import CORSMiddleware
13
+ import tensorflow as tf
14
+ tf.config.set_visible_devices([], 'GPU')
15
 
16
+ # Logging setelah model berhasil dimuat
17
+ logger.info("Loading model...")
18
  model = load_model_from_files("model.json", "my_model.h5")
19
+ logger.info("Model loaded successfully")
20
+
21
 
22
  labels = [
23
  "Benteng Vredeburg", "Candi Borobudur", "Candi Prambanan", "Gedung Agung Istana Kepresidenan",
 
93
  gr.Textbox(label="Deskripsi Lengkap", lines=20, max_lines=50),
94
  gr.HTML(label="Link Lokasi"),
95
  ],
96
+ flagging_mode="never",
97
  title="Klasifikasi Gambar",
98
  description="Upload gambar, sistem akan mengklasifikasikan dan memberikan deskripsi mengenai gambar tersebut."
99
  )
100
 
101
  # Mount Gradio ke FastAPI
102
+ app = gr.mount_gradio_app(app, gradio_app, path="/gradio")
103
 
104
  # Jalankan app
105
  if __name__ == "__main__":
requirements.txt CHANGED
@@ -3,3 +3,6 @@ keras
3
  tensorflow
4
  gradio
5
  torch
 
 
 
 
3
  tensorflow
4
  gradio
5
  torch
6
+ fastapi
7
+ uvicorn
8
+ pillow