Spaces:
Sleeping
Sleeping
Fixed error
Browse files- app.py +7 -8
- requirements.txt +1 -0
app.py
CHANGED
|
@@ -1,19 +1,13 @@
|
|
| 1 |
import os
|
| 2 |
-
import
|
| 3 |
-
from transformers import AutoModelForImageClassification, AutoFeatureExtractor
|
| 4 |
-
import torch
|
| 5 |
from flask_cors import CORS
|
| 6 |
-
from flask import Flask, request, json
|
| 7 |
-
import numpy as np
|
| 8 |
-
from PIL import Image
|
| 9 |
import requests
|
| 10 |
-
from io import BytesIO
|
| 11 |
from bs4 import BeautifulSoup
|
| 12 |
from urllib.parse import urljoin
|
| 13 |
import google.generativeai as genai
|
| 14 |
from langchain_google_genai import GoogleGenerativeAIEmbeddings
|
| 15 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
| 16 |
-
import warnings
|
| 17 |
from dotenv import load_dotenv
|
| 18 |
|
| 19 |
os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
|
@@ -62,6 +56,11 @@ def extract_images():
|
|
| 62 |
api_key=os.getenv("GOOGLE_API_KEY")
|
| 63 |
genai.configure(api_key=api_key)
|
| 64 |
model=genai.GenerativeModel('gemini-pro')
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 65 |
|
| 66 |
# @app.route("/predict", methods=["GET"])
|
| 67 |
# def predict():
|
|
|
|
| 1 |
import os
|
| 2 |
+
from transformers import pipeline
|
|
|
|
|
|
|
| 3 |
from flask_cors import CORS
|
| 4 |
+
from flask import Flask, request, json
|
|
|
|
|
|
|
| 5 |
import requests
|
|
|
|
| 6 |
from bs4 import BeautifulSoup
|
| 7 |
from urllib.parse import urljoin
|
| 8 |
import google.generativeai as genai
|
| 9 |
from langchain_google_genai import GoogleGenerativeAIEmbeddings
|
| 10 |
from langchain_google_genai import ChatGoogleGenerativeAI
|
|
|
|
| 11 |
from dotenv import load_dotenv
|
| 12 |
|
| 13 |
os.environ["CUDA_VISIBLE_DEVICES"] = ""
|
|
|
|
| 56 |
api_key=os.getenv("GOOGLE_API_KEY")
|
| 57 |
genai.configure(api_key=api_key)
|
| 58 |
model=genai.GenerativeModel('gemini-pro')
|
| 59 |
+
sentiment_analysis = pipeline("sentiment-analysis",model="siebert/sentiment-roberta-large-english")
|
| 60 |
+
|
| 61 |
+
# @app.route('/sentiment',methods=['POST'])
|
| 62 |
+
# def sentiment():
|
| 63 |
+
|
| 64 |
|
| 65 |
# @app.route("/predict", methods=["GET"])
|
| 66 |
# def predict():
|
requirements.txt
CHANGED
|
@@ -139,6 +139,7 @@ torch==2.2.1
|
|
| 139 |
torchmetrics==1.3.1
|
| 140 |
torchvision==0.17.1
|
| 141 |
tqdm==4.66.2
|
|
|
|
| 142 |
typing-inspect==0.9.0
|
| 143 |
typing_extensions==4.10.0
|
| 144 |
tzdata==2024.1
|
|
|
|
| 139 |
torchmetrics==1.3.1
|
| 140 |
torchvision==0.17.1
|
| 141 |
tqdm==4.66.2
|
| 142 |
+
transformers
|
| 143 |
typing-inspect==0.9.0
|
| 144 |
typing_extensions==4.10.0
|
| 145 |
tzdata==2024.1
|