NimaKL commited on
Commit
b7c66d4
·
verified ·
1 Parent(s): 483de8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -10,7 +10,6 @@ from sklearn.metrics.pairwise import cosine_similarity
10
 
11
  flask_app = Flask(__name__)
12
 
13
-
14
  class ModeratelySimplifiedGATConvModel(torch.nn.Module):
15
  def __init__(self, in_channels, hidden_channels, out_channels):
16
  super().__init__()
@@ -142,4 +141,4 @@ st.title("Video Recommendation API")
142
  st.write("Use POST requests to `/recommend` with JSON data {'input_text': '<your text>'}")
143
 
144
  if __name__ == "__main__":
145
- app.run(host='0.0.0.0', port=8501)
 
10
 
11
  flask_app = Flask(__name__)
12
 
 
13
  class ModeratelySimplifiedGATConvModel(torch.nn.Module):
14
  def __init__(self, in_channels, hidden_channels, out_channels):
15
  super().__init__()
 
141
  st.write("Use POST requests to `/recommend` with JSON data {'input_text': '<your text>'}")
142
 
143
  if __name__ == "__main__":
144
+ flask_app.run(host='0.0.0.0', port=8501)