mbesinci commited on
Commit
9f07d73
1 Parent(s): fab8302

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +46 -4
app.py CHANGED
@@ -1,7 +1,49 @@
1
  import gradio as gr
 
 
 
2
 
3
- def greet(name):
4
- return "Hello " + name + "!!"
5
 
6
- demo = gr.Interface(fn=greet, inputs="text", outputs="text")
7
- demo.launch()
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
+ import cv2
3
+ import torch
4
+ from transformers import AutoModelForImageTranslation # Bu, örnek bir model. Gerçek model ismini buraya yazmanız gerekebilir.
5
 
6
+ # Göz teması düzeltme modeli - model adını uygun bir modelle değiştirin
7
+ model = AutoModelForImageTranslation.from_pretrained("username/gaze_correction_model") # model yolunu doğru şekilde ayarlayın
8
 
9
+ # Video işlemi fonksiyonu
10
+ def correct_gaze_in_video(video_path):
11
+ # Video dosyasını aç
12
+ cap = cv2.VideoCapture(video_path)
13
+ fps = cap.get(cv2.CAP_PROP_FPS)
14
+ width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
15
+ height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
16
+
17
+ # Çıkış videosu için ayarlar
18
+ output_path = "corrected_gaze_output.mp4"
19
+ fourcc = cv2.VideoWriter_fourcc(*'mp4v')
20
+ out = cv2.VideoWriter(output_path, fourcc, fps, (width, height))
21
+
22
+ while cap.isOpened():
23
+ ret, frame = cap.read()
24
+ if not ret:
25
+ break
26
+
27
+ # Göz temasını düzeltmek için her kareyi işleme (örnek bir işleme işlemi)
28
+ inputs = preprocess_image(frame) # Görüntüyü modele uygun hale getir (normalize, resize vb.)
29
+ corrected_frame = model(inputs)
30
+
31
+ # Çıkışı video dosyasına ekleyin
32
+ out.write(corrected_frame)
33
+
34
+ # Kaynakları serbest bırak
35
+ cap.release()
36
+ out.release()
37
+
38
+ return output_path
39
+
40
+ # Gradio arayüzü
41
+ iface = gr.Interface(
42
+ fn=correct_gaze_in_video, # İşlem fonksiyonumuz
43
+ inputs="file", # Kullanıcıdan video dosyası girişi
44
+ outputs="file", # İşlenmiş video dosyası çıktısı
45
+ title="Gaze Correction in Video",
46
+ description="Upload a video to correct gaze direction."
47
+ )
48
+
49
+ iface.launch()