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
|