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)
|