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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -8,6 +8,9 @@ from torch_geometric.nn import GATConv
8
  from sentence_transformers import SentenceTransformer
9
  from sklearn.metrics.pairwise import cosine_similarity
10
 
 
 
 
11
  class ModeratelySimplifiedGATConvModel(torch.nn.Module):
12
  def __init__(self, in_channels, hidden_channels, out_channels):
13
  super().__init__()
@@ -126,13 +129,14 @@ if user_input:
126
 
127
 
128
 
129
- @app.route('/recommend', methods=['POST'])
130
  def recommend():
131
  input_text = request.json['input_text']
132
 
133
  recommendations = get_similar_and_recommend(input_text)
134
  return jsonify(recommendations)
135
 
 
136
  # Create a simple Streamlit interface with instructions
137
  st.title("Video Recommendation API")
138
  st.write("Use POST requests to `/recommend` with JSON data {'input_text': '<your text>'}")
 
8
  from sentence_transformers import SentenceTransformer
9
  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__()
 
129
 
130
 
131
 
132
+ @flask_app.route('/recommend', methods=['POST'])
133
  def recommend():
134
  input_text = request.json['input_text']
135
 
136
  recommendations = get_similar_and_recommend(input_text)
137
  return jsonify(recommendations)
138
 
139
+
140
  # Create a simple Streamlit interface with instructions
141
  st.title("Video Recommendation API")
142
  st.write("Use POST requests to `/recommend` with JSON data {'input_text': '<your text>'}")