lyly21 commited on
Commit
6cebde3
1 Parent(s): cb5bb87

Update mobilevit.html

Browse files
Files changed (1) hide show
  1. mobilevit.html +6 -7
mobilevit.html CHANGED
@@ -7,7 +7,7 @@
7
 
8
  <script type="module">
9
  // 허깅페이스의 pipeline 모듈을 import하십시오.
10
- // To-Do: ???
11
  // Make it available globally
12
  window.pipeline = pipeline;
13
  </script>
@@ -111,8 +111,8 @@
111
  let classifier;
112
  // Initialize the sentiment analysis model
113
  async function initializeModel() {
114
- // pipeline 함수를 이용하여 Xenova/mobilevit-small 모델의 인스턴스를 생성하여 이를 classifier에 저정하십시오. 인스턴스 생성 시 quantized 파라미터의 값을 false로 설정하십시오.
115
- // To-Do: ???
116
  }
117
  async function classifyImage() {
118
  const textFieldValue = document.getElementById("imageClassificationURLText").value.trim();
@@ -121,8 +121,7 @@
121
  }
122
  async function classifyImageLocal() {
123
  // HTML DOM의 element Id가 imageClassificationLocalFile인 element의 값을 fileInput으로 저장하십시오.
124
- // To-Do: const fileInput = ???
125
-
126
  const file = fileInput.files[0];
127
  if (!file) {
128
  alert('Please select an image file first.');
@@ -131,13 +130,13 @@
131
 
132
  const url = URL.createObjectURL(file);
133
  // classifier에 url을 입력하여 출력된 결과를 result에 저장하십시오.
134
- // To-Do: ???
135
  document.getElementById("outputAreaLocal").innerText = JSON.stringify(result, null, 2);
136
  }
137
  async function classifyTopImage() {
138
  const textFieldValue = document.getElementById("imageClassificationTopURLText").value.trim();
139
  // classifier에 textFieldValue를 입력 변수로, topk 파라미터 값을 3으로 설정하여 classifer를 수행하고 그 결과를 result에 저장하십시오.
140
- // To-Do: ???
141
  document.getElementById("outputAreaTop").innerText = JSON.stringify(result, null, 2);
142
  }
143
  // Initialize the model after the DOM is completely loaded
 
7
 
8
  <script type="module">
9
  // 허깅페이스의 pipeline 모듈을 import하십시오.
10
+ import { pipeline } from 'https://cdn.jsdelivr.net/npm/@xenova/[email protected]';
11
  // Make it available globally
12
  window.pipeline = pipeline;
13
  </script>
 
111
  let classifier;
112
  // Initialize the sentiment analysis model
113
  async function initializeModel() {
114
+ // pipeline 함수를 이용하여 Xenova/mobilevit-small 모델의 인스턴스를 생성하여 이를 classifier에 지정하십시오. 인스턴스 생성 시 quantized 파라미터의 값을 false로 설정하십시오.
115
+ classifier = pipeline('image-classification', { model: 'Xenova/mobilevit-small', quantized: false });
116
  }
117
  async function classifyImage() {
118
  const textFieldValue = document.getElementById("imageClassificationURLText").value.trim();
 
121
  }
122
  async function classifyImageLocal() {
123
  // HTML DOM의 element Id가 imageClassificationLocalFile인 element의 값을 fileInput으로 저장하십시오.
124
+ const fileInput = document.getElementById("imageClassificationLocalFile");
 
125
  const file = fileInput.files[0];
126
  if (!file) {
127
  alert('Please select an image file first.');
 
130
 
131
  const url = URL.createObjectURL(file);
132
  // classifier에 url을 입력하여 출력된 결과를 result에 저장하십시오.
133
+ const result = await classifier(url);
134
  document.getElementById("outputAreaLocal").innerText = JSON.stringify(result, null, 2);
135
  }
136
  async function classifyTopImage() {
137
  const textFieldValue = document.getElementById("imageClassificationTopURLText").value.trim();
138
  // classifier에 textFieldValue를 입력 변수로, topk 파라미터 값을 3으로 설정하여 classifer를 수행하고 그 결과를 result에 저장하십시오.
139
+ const result = await classifier(textFieldValue, {topK: 3});
140
  document.getElementById("outputAreaTop").innerText = JSON.stringify(result, null, 2);
141
  }
142
  // Initialize the model after the DOM is completely loaded