Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|