initial commit

Dockerfile

@@ -0,0 +1,9 @@
+FROM python:3.9-slim-buster
+COPY . /app
+WORKDIR /app
+RUN pip install --upgrade pip
+RUN pip install numpy==1.24.3
+RUN pip install -r requirements.txt
+EXPOSE 8501
+ENTRYPOINT ["streamlit","run"]
+CMD ["app.py"]

app.py

@@ -0,0 +1,41 @@
+import numpy as np
+import requests
+import streamlit as st
+
+def main():
+    st.title("Scientific Question Generation")
+
+    API_URL = "https://api-inference.huggingface.co/models/dhmeltzer/bart-large_askscience-qg"
+    headers = {"Authorization": "Bearer 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"}
+
+    def query(payload):
+        response = requests.post(API_URL, 
+                                 headers=headers, 
+                                 json=payload)
+        return response.json()
+    
+    # User search
+    user_input = st.text_area("Question Generator", 
+                              """Black holes are the most \
+                                gravitationally dense objects in the universe.""")
+
+    # Filters
+    st.sidebar.markdown("**Filters**")
+
+    temperature = st.sidebar.slider("Temperature", 0.0, 1.0, 0.0,.1)
+    num_results = st.sidebar.slider("Number of search results", 1, 50, 1)
+
+    vector = query([user_input])
+
+    if user_input:
+
+        output = query({
+	        "inputs": user_input,
+            "temperature":temperature,
+            "wait_for_model":True})
+
+        st.write(output)
+
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -0,0 +1,7 @@
+datasets
+torch==1.13
+transformers==4.29.0 
+faiss-cpu
+folium
+streamlit==1.14.0
+-e .

setup.py

@@ -0,0 +1,24 @@
+from setuptools import setup
+
+common_kwargs = dict(
+    version="0.1.0",
+    license="MIT",
+    author="David Meltzer",
+    author_email="davidhmeltzer@gmail.com",
+    classifiers=[
+        "Intended Audience :: Developers",
+        "Intended Audience :: Science/Research",
+        "License :: OSI Approved :: MIT License",
+        "Natural Language :: English",
+        "Operating System :: OS Independent",
+        "Programming Language :: Python :: 3.9",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    ],
+    python_requires=">=3.9",
+    include_package_data=False,
+)
+
+setup(
+    name="asks_qg",
+    **common_kwargs
+)