cantuncok commited on
Commit
4f12a0e
1 Parent(s): fd2dbe9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -8
app.py CHANGED
@@ -1,21 +1,33 @@
1
  import os
2
- from transformers import pipeline
3
  import gradio as gr
4
  from PIL import Image
 
5
 
6
- # Çevresel değişkenden tokeni al
7
  hf_token = os.getenv("HF_TOKEN")
8
 
9
- # Modeli yükleyin
10
  model_name = "cantuncok/autotrain1-model"
11
- classifier = pipeline("image-classification", model=model_name, use_auth_token=hf_token)
 
12
 
13
  # Görsel sınıflandırma fonksiyonu
14
  def classify_image(img):
15
- # Modelden tahminleri alın
16
- results = classifier(img)
17
- # Tahminleri ve olasılıklarını döndürün
18
- return {result['label']: result['score'] for result in results}
 
 
 
 
 
 
 
 
 
 
19
 
20
  # Gradio arayüzü için yapılandırma
21
  interface_options = {
 
1
  import os
2
+ from transformers import AutoModelForImageClassification, AutoProcessor
3
  import gradio as gr
4
  from PIL import Image
5
+ import torch
6
 
7
+ # Hugging Face tokeninizi çevresel değişkenden alın
8
  hf_token = os.getenv("HF_TOKEN")
9
 
10
+ # Model ve işlemciyi yükleyin
11
  model_name = "cantuncok/autotrain1-model"
12
+ processor = AutoProcessor.from_pretrained(model_name, use_auth_token=hf_token)
13
+ model = AutoModelForImageClassification.from_pretrained(model_name, use_auth_token=hf_token)
14
 
15
  # Görsel sınıflandırma fonksiyonu
16
  def classify_image(img):
17
+ # Görseli işlemci ile hazırlayın
18
+ inputs = processor(images=img, return_tensors="pt")
19
+ with torch.no_grad():
20
+ outputs = model(**inputs)
21
+
22
+ # Tahmin edilen sınıf indekslerini alın
23
+ logits = outputs.logits
24
+ predicted_class_idx = logits.argmax(-1).item()
25
+
26
+ # Etiket ve olasılık döndür
27
+ labels = model.config.id2label
28
+ predicted_label = labels[predicted_class_idx]
29
+ probabilities = torch.nn.functional.softmax(logits, dim=-1)
30
+ return {label: float(probabilities[0, idx]) for idx, label in labels.items()}
31
 
32
  # Gradio arayüzü için yapılandırma
33
  interface_options = {