zdou0830 commited on
Commit
551c110
1 Parent(s): 5390f50
Files changed (2) hide show
  1. main.py +1 -0
  2. requirements.txt +1 -0
main.py CHANGED
@@ -1,6 +1,7 @@
1
  import gradio as gr
2
  import torch
3
  import requests
 
4
  from torchvision import transforms
5
  import os
6
 
 
1
  import gradio as gr
2
  import torch
3
  import requests
4
+ import socket
5
  from torchvision import transforms
6
  import os
7
 
requirements.txt CHANGED
@@ -15,3 +15,4 @@ protobuf==3.20.1
15
  fastapi==0.74.*
16
  requests==2.27.*
17
  gradio
 
 
15
  fastapi==0.74.*
16
  requests==2.27.*
17
  gradio
18
+ socket