NimaKL commited on
Commit
bb0ac24
·
verified ·
1 Parent(s): 4a9119a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -0
app.py CHANGED
@@ -2,6 +2,7 @@ import streamlit as st
2
  import torch
3
  import pandas as pd
4
  import numpy as np
 
5
  from torch_geometric.data import Data
6
  from torch_geometric.nn import GATConv
7
  from sentence_transformers import SentenceTransformer
@@ -122,3 +123,19 @@ user_input = st.text_input("Enter text to find the most similar video")
122
  if user_input:
123
  recommendations = get_similar_and_recommend(user_input)
124
  st.json(recommendations)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  import torch
3
  import pandas as pd
4
  import numpy as np
5
+ from flask import Flask, request, jsonify
6
  from torch_geometric.data import Data
7
  from torch_geometric.nn import GATConv
8
  from sentence_transformers import SentenceTransformer
 
123
  if user_input:
124
  recommendations = get_similar_and_recommend(user_input)
125
  st.json(recommendations)
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>'}")
139
+
140
+ if __name__ == "__main__":
141
+ app.run(host='0.0.0.0', port=8501)