Remove streamlit

inference.py

@@ -1,10 +1,6 @@
-import streamlit as st
 import requests
-import csv
-import datetime
 import time
 import pandas as pd
-import matplotlib.pyplot as plt
 import os
 
 ##########################
@@ -13,11 +9,6 @@ headers = {"Authorization": "Bearer " + SECRET}
 API_URL = "https://api-inference.huggingface.co/models/cccmatthew/surrey-gp30"
 ##########################
 
-def log_to_csv(sentence, results, response_time):
-    with open('model_interactions.csv', 'a', newline='') as file:
-        writer = csv.writer(file)
-        writer.writerow([datetime.datetime.now(), sentence, results, response_time])
-
 def send_request_with_retry(url, headers, json_data, retries=3, backoff_factor=1):
     """Send request with retries on timeouts and HTTP 503 errors."""
     for attempt in range(retries):
@@ -29,14 +20,14 @@ def send_request_with_retry(url, headers, json_data, retries=3, backoff_factor=1
             return response, response_time
         except requests.exceptions.HTTPError as e:
             if response.status_code == 503:
-                st.info('Server is unavailable, retrying...')
+                print('Server is unavailable, retrying...')
             else:
                 raise
         except (requests.exceptions.ConnectionError, requests.exceptions.Timeout) as e:
-            st.info(f"Network issue ({str(e)}), retrying...")
+            print(f"Network issue ({str(e)}), retrying...")
         time.sleep(backoff_factor * (2 ** attempt))
 
-    st.error("Failed to process request after several attempts.")
+    print("Failed to process request after several attempts.")
     return None, None
 
 sentence = [
@@ -52,6 +43,7 @@ response, response_time = send_request_with_retry(API_URL, headers, {"inputs": s
 if response is not None:
     results = response.json()
     print(results)
+    response.close()
 
 else:
-    st.error("Unable to classify the sentence due to server issues.")
+    print("Unable to classify the sentence due to server issues.")